diff --git a/tests/integrations/database/model.py b/tests/integrations/database/model.py
index 536dacf..6fcb3f8 100644
--- a/tests/integrations/database/model.py
+++ b/tests/integrations/database/model.py
@@ -1,6 +1,3 @@
-import sys
-
-
 class LlmTrace(object):
     table_name = "llm_traces_preview_v01"