diff --git a/src/primaite/game/agent/observations/__init__.py b/src/primaite/game/agent/observations/__init__.py
index b9d97ae6..15fdf7ed 100644
--- a/src/primaite/game/agent/observations/__init__.py
+++ b/src/primaite/game/agent/observations/__init__.py
@@ -1,4 +1,5 @@
 # flake8: noqa
+# Pre-import all the observations when we load up the observations module so that they can be resolved by the parser.
 from primaite.game.agent.observations.acl_observation import ACLObservation
 from primaite.game.agent.observations.file_system_observations import FileObservation, FolderObservation
 from primaite.game.agent.observations.firewall_observation import FirewallObservation
@@ -10,3 +11,10 @@ from primaite.game.agent.observations.observation_manager import NestedObservati
 from primaite.game.agent.observations.observations import AbstractObservation
 from primaite.game.agent.observations.router_observation import RouterObservation
 from primaite.game.agent.observations.software_observation import ApplicationObservation, ServiceObservation
+
+# fmt: off
+__all__ = [
+    "ACLObservation", "FileObservation", "FolderObservation", "FirewallObservation", "HostObservation",
+    "LinksObservation", "NICObservation", "PortObservation", "NodesObservation", "NestedObservation",
+    "ObservationManager", "ApplicationObservation", "ServiceObservation",]
+# fmt: on