diff --git a/export_trt.py b/export_trt.py
index 3292e62..482303a 100644
--- a/export_trt.py
+++ b/export_trt.py
@@ -1,3 +1,6 @@
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 import torch
 import time
 from trt_utilities import Engine