diff --git a/src/primaite/game/agent/actions/__init__.py b/src/primaite/game/agent/actions/__init__.py
index 016a09ba..625725fe 100644
--- a/src/primaite/game/agent/actions/__init__.py
+++ b/src/primaite/game/agent/actions/__init__.py
@@ -14,7 +14,6 @@ from primaite.game.agent.actions import (
     service,
     session,
 )
-from primaite.game.agent.actions.manager import ActionManager
 
 __all__ = (
     "abstract",
@@ -29,5 +28,4 @@ __all__ = (
     "node",
     "service",
     "session",
-    "ActionManager",
 )
diff --git a/src/primaite/game/agent/actions/manager.py b/src/primaite/game/agent/actions/manager.py
index 6c8353b0..3795d21d 100644
--- a/src/primaite/game/agent/actions/manager.py
+++ b/src/primaite/game/agent/actions/manager.py
@@ -22,9 +22,7 @@ from gymnasium import spaces
 from primaite.game.agent.actions.abstract import AbstractAction
 from primaite.interface.request import RequestFormat
 
-# TODO: Make sure that actions are backwards compatible where the old YAML format is used.
-
-__all__ = "DoNothingAction"
+__all__ = ("DoNothingAction", "ActionManager")
 
 
 class DoNothingAction(AbstractAction, identifier="do_nothing"):