diff --git a/NeoBoot/tmpfiles/runpy/mips_run.py b/NeoBoot/tmpfiles/runpy/mips_run.py
index 3c82197..8813435 100644
--- a/NeoBoot/tmpfiles/runpy/mips_run.py
+++ b/NeoBoot/tmpfiles/runpy/mips_run.py
@@ -134,11 +134,14 @@ class StartImage(Screen):
                             self.close()
                             
 def getREBOOTSYSTEM():                
-                for line in open("/etc/hostname"):                                
+    __module__ = __name__
+
+    def __init__(self, session):
+        Screen.__init__(self, session)                
+        for line in open("/etc/hostname"):                                
                         if "dm500hd" in line or "dm800se" in line or "dm800" in line or "dm800se" in line or "dm8000" in line:
                                 if fileExists('%sImageBoot/%s/squashfs-images' % (getNeoLocation(), getImageNeoBoot())):
                                     os.system('ln -sf "%sImageBoot/%s/squashfs-images" "//squashfs-images"' % (getNeoLocation(), getImageNeoBoot()))
                                 os.system('sleep 5;reboot -f; shutdown -r now &')
                         else:
-                                os.system('sleep 5;reboot -f; shutdown -r now &')
-                                #self.session.open(TryQuitMainloop, 2)
+                                self.session.open(TryQuitMainloop, 2)