diff --git a/src/primaite/game/agent/interface.py b/src/primaite/game/agent/interface.py
index aac6c05a..d5165a71 100644
--- a/src/primaite/game/agent/interface.py
+++ b/src/primaite/game/agent/interface.py
@@ -38,7 +38,7 @@ class AgentHistoryItem(BaseModel):
 
     reward_info: Dict[str, Any] = {}
 
-    obs_space_data: Optional[ObsType] = None
+    observation: Optional[ObsType] = None
     """The observation space data for this step."""
 
 
@@ -178,7 +178,7 @@ class AbstractAgent(ABC):
         parameters: Dict[str, Any],
         request: RequestFormat,
         response: RequestResponse,
-        obs_space_data: ObsType,
+        observation: ObsType,
     ) -> None:
         """Process the response from the most recent action."""
         self.history.append(
@@ -188,7 +188,7 @@ class AbstractAgent(ABC):
                 parameters=parameters,
                 request=request,
                 response=response,
-                obs_space_data=obs_space_data,
+                observation=observation,
             )
         )
 
diff --git a/src/primaite/game/game.py b/src/primaite/game/game.py
index ed3c84d3..4f21120d 100644
--- a/src/primaite/game/game.py
+++ b/src/primaite/game/game.py
@@ -186,7 +186,7 @@ class PrimaiteGame:
                 parameters=parameters,
                 request=request,
                 response=response,
-                obs_space_data=obs,
+                observation=obs,
             )
 
     def pre_timestep(self) -> None:
diff --git a/tests/integration_tests/game_layer/observations/test_obs_data_capture.py b/tests/integration_tests/game_layer/observations/test_obs_data_capture.py
index 810b2ad7..e8bdea22 100644
--- a/tests/integration_tests/game_layer/observations/test_obs_data_capture.py
+++ b/tests/integration_tests/game_layer/observations/test_obs_data_capture.py
@@ -25,4 +25,4 @@ def test_obs_data_in_log_file():
     with open(path, "r") as f:
         j = json.load(f)
 
-    assert type(j["0"]["defender"]["obs_space_data"]) == dict
+    assert type(j["0"]["defender"]["observation"]) == dict