diff --git a/sdk/src/og_sdk/utils.py b/sdk/src/og_sdk/utils.py
index a6757b0..6489274 100644
--- a/sdk/src/og_sdk/utils.py
+++ b/sdk/src/og_sdk/utils.py
@@ -6,6 +6,7 @@
 import string
 import random
 import aiofiles
+import logging
 from og_proto import agent_server_pb2, common_pb2
 from typing import AsyncIterable
 
diff --git a/sdk/tests/agent_sdk_tests.py b/sdk/tests/agent_sdk_tests.py
index a898f1c..8e98629 100644
--- a/sdk/tests/agent_sdk_tests.py
+++ b/sdk/tests/agent_sdk_tests.py
@@ -22,6 +22,7 @@
 import logging
 import json
 import random
+import logging
 from tempfile import gettempdir
 from pathlib import Path
 from og_sdk.agent_sdk import AgentSDK