diff --git a/src/clean_agent_preview.py b/src/clean_agent_preview.py
index 4781dcc3b..5d7d35f18 100644
--- a/src/clean_agent_preview.py
+++ b/src/clean_agent_preview.py
@@ -6,6 +6,7 @@ import base64
 import importlib.util
 import io
 import json
+import logging
 import os
 import re
 import sys