diff --git a/src/snapshot_page/new_snapshot_dialog.py b/src/snapshot_page/new_snapshot_dialog.py
index 660b21d..e2fbdda 100644
--- a/src/snapshot_page/new_snapshot_dialog.py
+++ b/src/snapshot_page/new_snapshot_dialog.py
@@ -97,6 +97,7 @@ class NewSnapshotDialog(Adw.Dialog):
         return True
         
     def on_create(self, button):
+        self.snapshot_page.status_stack.set_visible_child(self.snapshot_page.snapshotting_view)
         self.workers.clear()
         for row in self.selected_rows:
             if "io.github.flattool.Warehouse" in row.package.info["id"]:
@@ -138,10 +139,11 @@ class NewSnapshotDialog(Adw.Dialog):
         if self.create_button.get_sensitive():
             self.create_button.activate()
     
-    def __init__(self, parent_page, loading_status, on_done=None, package=None, **kwargs):
+    def __init__(self, snapshot_page, loading_status, on_done=None, package=None, **kwargs):
         super().__init__(**kwargs)
         
         # Extra Object Creations
+        self.snapshot_page = snapshot_page
         self.loading_status = loading_status
         self.on_done = on_done
         self.rows = []
diff --git a/src/snapshot_page/snapshot_page.py b/src/snapshot_page/snapshot_page.py
index e225627..9511a3b 100644
--- a/src/snapshot_page/snapshot_page.py
+++ b/src/snapshot_page/snapshot_page.py
@@ -151,7 +151,6 @@ class SnapshotPage(Adw.BreakpointBin):
     def end_loading(self):
         def callback(*args):
             self.new_snapshot_dialog = NewSnapshotDialog(self, self.snapshotting_status, self.refresh)
-            self.new_snapshot_dialog.create_button.connect("clicked", lambda *_: self.status_stack.set_visible_child(self.snapshotting_view))
             self.generate_active_list()
             self.generate_leftover_list()
             if (not self.active_box.get_visible()) and (not self.leftover_box.get_visible()):
diff --git a/src/snapshot_page/snapshots_list_page.py b/src/snapshot_page/snapshots_list_page.py
index 8b314ea..1015230 100644
--- a/src/snapshot_page/snapshots_list_page.py
+++ b/src/snapshot_page/snapshots_list_page.py
@@ -72,7 +72,6 @@ class SnapshotsListPage(Adw.NavigationPage):
             
     def on_new(self, button):
         dialog = NewSnapshotDialog(self.parent_page, self.parent_page.snapshotting_status, self.on_done, self.package_or_folder)
-        dialog.create_button.connect("clicked", lambda *_: self.parent_page.status_stack.set_visible_child(self.parent_page.snapshotting_view))
         dialog.present(HostInfo.main_window)
 
     def sort_func(self, row1, row2):