diff --git a/lib/livebook_web/live/session_live/persistence_live.ex b/lib/livebook_web/live/session_live/persistence_live.ex
index baa680c0c..70017d105 100644
--- a/lib/livebook_web/live/session_live/persistence_live.ex
+++ b/lib/livebook_web/live/session_live/persistence_live.ex
@@ -91,10 +91,12 @@ defmodule LivebookWeb.SessionLive.PersistenceLive do
                 <span class="text-gray-700 whitespace-no-wrap">
                   no file selected
                 </span>
-                <button class="button-base button-gray button-small"
-                  phx-click="open_file_select">
-                  Choose a file
-                </button>
+                <%= unless @draft_file do %>
+                  <button class="button-base button-gray button-small"
+                    phx-click="open_file_select">
+                    Choose a file
+                  </button>
+                <% end %>
               <% end %>
             </div>
           </form>