diff --git a/src/local_search.py b/src/local_search.py
index dc8669b..7399f6d 100644
--- a/src/local_search.py
+++ b/src/local_search.py
@@ -1,4 +1,5 @@
 import os
+import numpy as np
 if 'AI' in os.getcwd():
     from src.utils import *
 else: