diff --git a/graphiti_core/driver/driver.py b/graphiti_core/driver/driver.py
index a8d94022..40ce2540 100644
--- a/graphiti_core/driver/driver.py
+++ b/graphiti_core/driver/driver.py
@@ -21,7 +21,7 @@ from abc import ABC, abstractmethod
 from collections.abc import Coroutine
 from datetime import datetime
 from enum import Enum
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 from graphiti_core.embedder.client import EMBEDDING_DIM