diff --git a/assets/src/ba_data/python/ba/_coopsession.py b/assets/src/ba_data/python/ba/_coopsession.py
index fed532e9..bb85a4b0 100644
--- a/assets/src/ba_data/python/ba/_coopsession.py
+++ b/assets/src/ba_data/python/ba/_coopsession.py
@@ -90,7 +90,7 @@ class CoopSession(Session):
         """Get the game instance currently being played."""
         return self._current_game_instance
 
-    def should_allow_mid_activity_joins(self, activity: Activity) -> bool:
+    def should_allow_mid_activity_joins(self, activity: ba.Activity) -> bool:
         # pylint: disable=cyclic-import
         from ba._gameactivity import GameActivity
 
diff --git a/assets/src/ba_data/python/ba/_session.py b/assets/src/ba_data/python/ba/_session.py
index 2c7e6520..9170d214 100644
--- a/assets/src/ba_data/python/ba/_session.py
+++ b/assets/src/ba_data/python/ba/_session.py
@@ -205,7 +205,7 @@ class Session:
             raise NodeNotFoundError()
         return node
 
-    def should_allow_mid_activity_joins(self, activity: Activity) -> bool:
+    def should_allow_mid_activity_joins(self, activity: ba.Activity) -> bool:
         """Returned value is used by the Session to determine
         whether to allow players to join in the middle of activity.