diff --git a/erynrl/ai.py b/erynrl/ai.py
index 710b2b1..05c761e 100644
--- a/erynrl/ai.py
+++ b/erynrl/ai.py
@@ -7,7 +7,7 @@ import numpy as np
 import tcod
 
 from . import log
-from .actions import Action
+from .actions.action import Action
 from .actions.game import BumpAction, WaitAction
 from .components import Component
 from .geometry import Direction, Point