diff --git a/agents/nec_agent.py b/agents/nec_agent.py
index 77520a4..a327db4 100644
--- a/agents/nec_agent.py
+++ b/agents/nec_agent.py
@@ -15,7 +15,7 @@
 #
 
 import numpy as np
-
+import os, pickle
 from agents.value_optimization_agent import ValueOptimizationAgent
 from logger import screen
 from utils import RunPhase