diff --git a/desktop_env/controllers/python.py b/desktop_env/controllers/python.py
index 667658d..d4391c8 100644
--- a/desktop_env/controllers/python.py
+++ b/desktop_env/controllers/python.py
@@ -3,6 +3,7 @@ import logging
 import random
 from typing import Any, Dict, Optional
 import time
+import traceback
 import requests
 
 from desktop_env.actions import KEYBOARD_KEYS