diff --git a/config.py b/config.py
index 9c0e689..69506ed 100644
--- a/config.py
+++ b/config.py
@@ -24,6 +24,5 @@ config = {
         'update': {'enabled': True},
         'data_privacy': {'enabled': True},
         'term_privacy': {'enabled': True}
-    },
-    'UPDATE_BRANCH': 'dev-rework'
+    }
 }
diff --git a/main.py b/main.py
index 2139129..ce54ad8 100644
--- a/main.py
+++ b/main.py
@@ -61,8 +61,8 @@ class Selena(discord.Client):
 
         if config['modules']['update']['enabled']:
             from modules.admin.update import setup as update_setup
-            branch_name = config.get('UPDATE_BRANCH', 'dev-rework')
-            update_setup(self, branch=branch_name)
+            update_setup = update_setup(self)
+            update_setup(self.tree)
 
         if config['modules']['data_privacy']['enabled']:
             from modules.admin.data_privacy import DataPrivacy