diff --git a/qt_ui/windows/basemenu/QBaseMenu2.py b/qt_ui/windows/basemenu/QBaseMenu2.py
index 4206c543..838da2ca 100644
--- a/qt_ui/windows/basemenu/QBaseMenu2.py
+++ b/qt_ui/windows/basemenu/QBaseMenu2.py
@@ -171,7 +171,7 @@ class QBaseMenu2(QDialog):
     def new_package(self) -> None:
         Dialog.open_new_package_dialog(self.cp, parent=self.window())
 
-    def update_dialogue_budget(self, budget):
+    def update_dialogue_budget(self, budget: int):
         GameUpdateSignal.get_instance().updateBudget(self.game_model.game)
         for child in self.children():
             if child.objectName() == "budgetField":