diff --git a/tests/test_integration.py b/tests/test_integration.py
index 310f844..9b6fb6a 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -1,7 +1,6 @@
 import pytest
 from exchange.exchange import Exchange
 from exchange.message import Message
-from exchange.moderators import ContextTruncate
 from exchange.providers import get_provider
 from exchange.tool import Tool
 from tests import openai_model_tool
diff --git a/tests/test_integration_vision.py b/tests/test_integration_vision.py
index e28276a..5a03df1 100644
--- a/tests/test_integration_vision.py
+++ b/tests/test_integration_vision.py
@@ -2,7 +2,6 @@
 from exchange.content import ToolResult, ToolUse
 from exchange.exchange import Exchange
 from exchange.message import Message
-from exchange.moderators import ContextTruncate
 from exchange.providers import get_provider
 from tests import openai_model_vision