diff --git a/titles/idac/season2.py b/titles/idac/season2.py
index d948534..e32706c 100644
--- a/titles/idac/season2.py
+++ b/titles/idac/season2.py
@@ -2493,7 +2493,7 @@ class IDACSeason2(IDACBase):
             },
         )
 
-        vs_info = _update_vs_info(user_id, 0, data)
+        vs_info = self._update_vs_info(user_id, IDACConstants.BATTLE_MODE_ONLINE, data)
 
         return {
             "status_code": "0",
@@ -2572,7 +2572,7 @@ class IDACSeason2(IDACBase):
         )
 
         # save vs_info in database
-        vs_info = _update_vs_info(user_id, 1, data)
+        vs_info = self._update_vs_info(user_id, IDACConstants.BATTLE_MODE_OFFLINE, data)
 
         return {
             "status_code": "0",