diff --git a/etc/unittest/mocks.py b/etc/unittest/mocks.py
index 50d1a5a4..a903ebfd 100644
--- a/etc/unittest/mocks.py
+++ b/etc/unittest/mocks.py
@@ -1,5 +1,5 @@
 from g4f.providers.base_provider import AbstractProvider, AsyncProvider, AsyncGeneratorProvider
-from g4f.image import ImageResponse
+from g4f.providers.response import ImageResponse
 from g4f.errors import MissingAuthError
 
 class ProviderMock(AbstractProvider):
diff --git a/g4f/requests/__init__.py b/g4f/requests/__init__.py
index d1f5d024..88b8a47a 100644
--- a/g4f/requests/__init__.py
+++ b/g4f/requests/__init__.py
@@ -137,7 +137,7 @@ async def get_nodriver(
     timeout: int = 120,
     browser_executable_path=None,
     **kwargs
-) -> Browser:
+) -> tuple[Browser, callable]:
     if not has_nodriver:
         raise MissingRequirementsError('Install "nodriver" and "platformdirs" package | pip install -U nodriver platformdirs')
     user_data_dir = user_config_dir(f"g4f-{user_data_dir}") if has_platformdirs else None