diff --git a/main.py b/main.py
index 8e7fc47..ca52b77 100644
--- a/main.py
+++ b/main.py
@@ -579,7 +579,10 @@ if __name__ == "__main__":
         except Exception as e:
             print(f"Error starting client: {e}", file=sys.stderr)
             if isinstance(e, sqlite3.OperationalError) and "database is locked" in str(e):
-                print("Database lock detected. Please ensure no other instances are running.", file=sys.stderr)
+                print(
+                    "Database lock detected. Please ensure no other instances are running.",
+                    file=sys.stderr
+                )
             sys.exit(1)
 
     asyncio.run(main())
diff --git a/tests/test_basic.py b/tests/test_basic.py
index 38fead6..4692943 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -24,8 +24,6 @@ def mock_client():
 def test_imports():
     """Test that all necessary imports are available"""
     # This will fail if any import is missing
-    import main
-    from mcp.server.fastmcp import FastMCP
     from telethon import TelegramClient
     from telethon.sessions import StringSession