diff --git a/lbforaging/agents/heuristic_agent.py b/lbforaging/agents/heuristic_agent.py
index 872d1be..b3c8c2d 100644
--- a/lbforaging/agents/heuristic_agent.py
+++ b/lbforaging/agents/heuristic_agent.py
@@ -3,7 +3,7 @@
 import numpy as np
 
 from lbforaging.agents.agent import BaseAgent
-from foraging.environment import Action
+from lbforaging.foraging.environment import Action
 
 
 class HeuristicAgent(BaseAgent):