diff --git a/src/view/code.tsx b/src/view/code.tsx
index a00d381f..d1c4823b 100644
--- a/src/view/code.tsx
+++ b/src/view/code.tsx
@@ -203,6 +203,9 @@ class SourceCodeRenderer extends React.Component<
                     this.setState({ message: null });
                 }, 3000);
             });
+        if (this.props.shouldFocus) {
+            GlobalCommandRunner.screenSetFocus("input");
+        }
     };
 
     handleEditorChange = (code) => {