From 3a2d7d4817b3be3b80f48c632120a7b83e375c01 Mon Sep 17 00:00:00 2001
From: Janne Valkealahti <janne.valkealahti@gmail.com>
Date: Sat, 22 Jul 2023 17:27:21 +0100
Subject: [PATCH] Restore focus after returning from scenario

- Relates #811
---
 .../java/org/springframework/shell/samples/catalog/Catalog.java  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/spring-shell-samples/spring-shell-sample-catalog/src/main/java/org/springframework/shell/samples/catalog/Catalog.java b/spring-shell-samples/spring-shell-sample-catalog/src/main/java/org/springframework/shell/samples/catalog/Catalog.java
index 46b7cd34..93a19764 100644
--- a/spring-shell-samples/spring-shell-sample-catalog/src/main/java/org/springframework/shell/samples/catalog/Catalog.java
+++ b/spring-shell-samples/spring-shell-sample-catalog/src/main/java/org/springframework/shell/samples/catalog/Catalog.java
@@ -127,6 +127,7 @@ public class Catalog {
 					if (currentScenarioView != null) {
 						currentScenarioView = null;
 						ui.setRoot(app, true);
+						ui.setFocus(categories);
 					}
 					else {
 						requestQuit();