diff --git a/run/train.py b/run/train.py
index ad49ca4..2e732dd 100644
--- a/run/train.py
+++ b/run/train.py
@@ -4,9 +4,10 @@
 from copy import deepcopy
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
-from run.grid_search import GridSearch
+from utility.grid_search import GridSearch
 from utility.yaml_op import load_args
 from utility.debug_tools import assert_colorize
+from utility.utils import str2bool
 
 
 def parse_cmd_args():
diff --git a/run/grid_search.py b/utility/grid_search.py
similarity index 100%
rename from run/grid_search.py
rename to utility/grid_search.py