diff --git a/framework/python/src/core/session.py b/framework/python/src/core/session.py
index f65da0bb5..c3fbc83c5 100644
--- a/framework/python/src/core/session.py
+++ b/framework/python/src/core/session.py
@@ -635,7 +635,7 @@ def update_profile(self, profile_json):
 
     return risk_profile
 
-  def _remove_invalid_questions(self, questions: list[dict]) -> list[dict]:
+  def _remove_invalid_questions(self, questions):
     """Remove unrecognised questions from the profile"""
 
     # Store valid questions
diff --git a/make/DEBIAN/control b/make/DEBIAN/control
index 31922abde..c822e65fd 100644
--- a/make/DEBIAN/control
+++ b/make/DEBIAN/control
@@ -1,5 +1,5 @@
 Package: Testrun
-Version: 2.0
+Version: 2.0.1
 Architecture: amd64
 Maintainer: Google <boddey@google.com>
 Homepage: https://github.com/google/testrun