diff --git a/YesNt.CodeEditor/InputHandler.cs b/YesNt.CodeEditor/InputHandler.cs
index b937493..55dd583 100644
--- a/YesNt.CodeEditor/InputHandler.cs
+++ b/YesNt.CodeEditor/InputHandler.cs
@@ -50,10 +50,12 @@ internal class InputHandler(TextEditor textEditor)
 
                     case ConsoleKey.R:
                         ExecuteWithDebugScreen("run", false, false);
+                        textEditor.Display(true);
                         return true;
 
                     case ConsoleKey.D:
                         ExecuteWithDebugScreen("debug", true, false);
+                        textEditor.Display(true);
                         return true;
 
                     case ConsoleKey.F: