diff --git a/Interface/AppUI.py b/Interface/AppUI.py
index 1eca962..289e2ef 100644
--- a/Interface/AppUI.py
+++ b/Interface/AppUI.py
@@ -58,10 +58,6 @@ def setSender(self, new_sender: ISender):
     def setUser(self, current_user: User):
         self.user = current_user
 
-    def setDb(self, new_db):
-        #AppUI.db = new_db
-        IModel.db = new_db
-
     def add_periodic_task(self, period: int, func: Callable):
         # TODO można poprawić żeby się odpalało tylko przy dodaniu obiektu,
         # przemyśleć
diff --git a/main.py b/main.py
index 14c4ac0..21992ef 100644
--- a/main.py
+++ b/main.py
@@ -31,6 +31,7 @@ def on_closing():
 if __name__ == "__main__":
     db = DatabaseHandler(dbURL, tables)
     GroupController.setDbHandler(db)
+    IModel.db = db
     
     if db.checkIntegrity():
         print("Database intact, proceeding.")
@@ -41,7 +42,6 @@ def on_closing():
     
     ui = AppUI()
     ui.prepareInterface()
-    ui.setDb(db)
     
     
     if (mocking_enabled):