diff --git a/index.html b/index.html
index 8009c4e..68956a4 100644
--- a/index.html
+++ b/index.html
@@ -113,6 +113,7 @@ By <a href="mailto:inexorabletash@gmail.com">Joshua Bell</a>
   </textarea>
   <button id="btn_save" title="Save">&#x1f4be;</button>
   <button id="btn_load" title="Load">&#x1f4c2;</button>
+
   <input type="submit" id="btn_share" value="Share your sample!">
   <input type="button" id="show_paper" value="Show output" title="Echo all output to a &quot;print-out&quot;, so you can copy/paste">
   <input type="button" id="hide_paper" value="Hide output" title="Hide the &quot;print-out&quot;">
diff --git a/index.js b/index.js
index ee5ca68..21115c0 100644
--- a/index.js
+++ b/index.js
@@ -207,7 +207,7 @@ window.addEventListener('DOMContentLoaded', function () {
     $("#btn_run").disabled = stopped ? "" : "disabled";
     $("#lb_files").disabled = stopped ? "" : "disabled";
 
-    if (btnFocus) {
+    if (btnFocus || stopped) {
       $(stopped ? "#btn_run" : "#btn_stop").focus();
     } else {
       tty.focus();