diff --git a/src/primaite/game/agent/observations/agent_observations.py b/src/primaite/game/agent/observations/agent_observations.py
index 2148697b..10370660 100644
--- a/src/primaite/game/agent/observations/agent_observations.py
+++ b/src/primaite/game/agent/observations/agent_observations.py
@@ -2,7 +2,7 @@ from typing import Dict, List, Optional, Tuple, TYPE_CHECKING
 
 from gymnasium import spaces
 
-from primaite.game.agent.observations.host import NodeObservation
+from primaite.game.agent.observations.node_observations import NodeObservation
 from primaite.game.agent.observations.observations import (
     AbstractObservation,
     AclObservation,
diff --git a/tests/integration_tests/game_layer/observations/test_node_observations.py b/tests/integration_tests/game_layer/observations/test_node_observations.py
index a5195e1e..dce05b6a 100644
--- a/tests/integration_tests/game_layer/observations/test_node_observations.py
+++ b/tests/integration_tests/game_layer/observations/test_node_observations.py
@@ -4,7 +4,7 @@ from uuid import uuid4
 import pytest
 from gymnasium import spaces
 
-from primaite.game.agent.observations.host import NodeObservation
+from primaite.game.agent.observations.node_observations import NodeObservation
 from primaite.simulator.network.hardware.nodes.host.computer import Computer
 from primaite.simulator.sim_container import Simulation