diff --git a/models/object_detection/code/coco_object_detection.py b/models/object_detection/code/coco_object_detection.py
index cf5830bd8..0d9888529 100644
--- a/models/object_detection/code/coco_object_detection.py
+++ b/models/object_detection/code/coco_object_detection.py
@@ -10,6 +10,8 @@
 import cv2
 from PIL import Image
 
+import traceback
+
 def main():
     os.chdir('/home/pi/DeepPiCar/models/object_detection')
     
diff --git a/models/object_detection/code/object_detection_usb.py b/models/object_detection/code/object_detection_usb.py
index 477b5b772..d54f55f6d 100644
--- a/models/object_detection/code/object_detection_usb.py
+++ b/models/object_detection/code/object_detection_usb.py
@@ -10,6 +10,8 @@
 import cv2
 from PIL import Image
 
+import traceback
+
 def main():
     os.chdir('/home/pi/DeepPiCar/models/object_detection')