diff --git a/agent.py b/agent.py
index 25e5f7f..4db5b90 100755
--- a/agent.py
+++ b/agent.py
@@ -9,6 +9,7 @@ import sys
 import pdb
 import numpy as np
 from package_location_classifier.classifier import PackageLocationClassifier
+from genetic_route import genetic_trace_route
 
 class Node:
 	def __init__(self, coord_x, coord_y, package=None, is_rack=False):