diff --git a/eppo_client/client.py b/eppo_client/client.py
index cc7852c..4ae6f1f 100644
--- a/eppo_client/client.py
+++ b/eppo_client/client.py
@@ -339,7 +339,7 @@ def get_bandit_action_detail(
             subject_key,
             subject_attributes,
             action_contexts,
-            bandit_data.model_data,
+            bandit_data.bandit_model_data,
         )
 
         # log bandit action
@@ -350,7 +350,7 @@ def get_bandit_action_detail(
             "action": evaluation.action_key if evaluation else None,
             "actionProbability": evaluation.action_weight if evaluation else None,
             "optimalityGap": evaluation.optimality_gap if evaluation else None,
-            "modelVersion": bandit_data.model_version if evaluation else None,
+            "modelVersion": bandit_data.bandit_model_version if evaluation else None,
             "timestamp": datetime.datetime.utcnow().isoformat(),
             "subjectNumericAttributes": (
                 subject_attributes.numeric_attributes
diff --git a/eppo_client/models.py b/eppo_client/models.py
index ea4e75d..54add2f 100644
--- a/eppo_client/models.py
+++ b/eppo_client/models.py
@@ -1,5 +1,6 @@
 from datetime import datetime
 from enum import Enum
+from pydantic import Field
 from typing import Dict, List, Optional
 
 from eppo_client.base_model import SdkBaseModel
@@ -96,7 +97,7 @@ class BanditModelData(SdkBaseModel):
 
 class BanditData(SdkBaseModel):
     bandit_key: str
-    model_name: str
+    bandit_model_name: str = Field(alias="modelName")
+    bandit_model_version: str = Field(alias="modelVersion")
+    bandit_model_data: BanditModelData = Field(alias="modelData")
     updated_at: datetime
-    model_version: str
-    model_data: BanditModelData
diff --git a/eppo_client/version.py b/eppo_client/version.py
index 88c513e..903a158 100644
--- a/eppo_client/version.py
+++ b/eppo_client/version.py
@@ -1 +1 @@
-__version__ = "3.3.0"
+__version__ = "3.4.0"
diff --git a/test/client_bandit_test.py b/test/client_bandit_test.py
index a3176ae..64cb7c6 100644
--- a/test/client_bandit_test.py
+++ b/test/client_bandit_test.py
@@ -175,9 +175,9 @@ def test_bandit_generic_test_cases(test_case):
             flag,
             subject["subjectKey"],
             ContextAttributes(
-                numeric_attributes=subject["subjectAttributes"]["numeric_attributes"],
+                numeric_attributes=subject["subjectAttributes"]["numericAttributes"],
                 categorical_attributes=subject["subjectAttributes"][
-                    "categorical_attributes"
+                    "categoricalAttributes"
                 ],
             ),
             {