diff --git a/mass_storage/main.py b/mass_storage/main.py
index d4211a4..9cdae55 100644
--- a/mass_storage/main.py
+++ b/mass_storage/main.py
@@ -6,7 +6,7 @@ ___dependencies___ = ["dialogs", "ugfx_helper"]
 ___categories___   = ["EMF"]
 ___bootstrapped___ = True
 
-import ugfx, tilda, ugfx_helper, dialogs
+import ugfx, tilda, ugfx_helper, dialogs, app
 
 ugfx_helper.init()
 ugfx.clear()
@@ -18,3 +18,4 @@ with dialogs.WaitingMessage(title="Mass Storage Enabled", text="You can now use
     print("Waiting for USB mass storage to be unmounted...")
     tilda.storage_disable_usb()
     print("DONE")
+    app.restart_to_default()