diff --git a/agentops/client.py b/agentops/client.py
index 65608e0e..67c0dbfa 100644
--- a/agentops/client.py
+++ b/agentops/client.py
@@ -119,8 +119,7 @@ def _check_for_partner_frameworks(self, instrument_llm_calls, auto_start_session
                 if framework == 'autogen':
                     try:
                         import autogen
-                        from .partners.autogen_logger import AutogenLogger  # TODO: move?
-                        # TODO: can we pass it like this without it being a singleton?
+                        from .partners.autogen_logger import AutogenLogger
                         autogen.runtime_logging.start(logger=AutogenLogger())
                     except ImportError:
                         pass