diff --git a/src/game/engine/scenes/hud/chronicle_panel.gd b/src/game/engine/scenes/hud/chronicle_panel.gd
index 28c8b256..3444956b 100644
--- a/src/game/engine/scenes/hud/chronicle_panel.gd
+++ b/src/game/engine/scenes/hud/chronicle_panel.gd
@@ -332,7 +332,7 @@ func _request_frame_render() -> void:
 	var game_map: RefCounted = GameState.get_game_map()
 	if game_map == null:
 		return
-	var store: GdObservationStore = GameState.get_observation_store(
+	var store: RefCounted = GameState.get_observation_store(
 		_current_player_index, game_map.width, game_map.height
 	)
 	if store == null:
diff --git a/src/game/engine/scenes/world_map/world_map_vision.gd b/src/game/engine/scenes/world_map/world_map_vision.gd
index a07432b3..a5773a88 100644
--- a/src/game/engine/scenes/world_map/world_map_vision.gd
+++ b/src/game/engine/scenes/world_map/world_map_vision.gd
@@ -30,7 +30,7 @@ static func record_observations(player: RefCounted, game_map: RefCounted) -> voi
 	var grid: RefCounted = GameState.get("_grid_state")
 	if grid == null:
 		return
-	var store: GdObservationStore = GameState.get_observation_store(
+	var store: RefCounted = GameState.get_observation_store(
 		player.index, game_map.width, game_map.height
 	)
 	if store == null: