diff --git a/app/logger.py b/app/logger.py
index 022e67297..cdfb862cf 100644
--- a/app/logger.py
+++ b/app/logger.py
@@ -46,8 +46,10 @@ def get_logs():
 
 
 def on_flush(callback):
-    stdout_interceptor.on_flush(callback)
-    stderr_interceptor.on_flush(callback)
+    if stdout_interceptor is not None:
+        stdout_interceptor.on_flush(callback)
+    if stderr_interceptor is not None:
+        stderr_interceptor.on_flush(callback)
 
 def setup_logger(log_level: str = 'INFO', capacity: int = 300):
     global logs
diff --git a/tests-unit/server/routes/internal_routes_test.py b/tests-unit/server/routes/internal_routes_test.py
index 862f89ed0..4fe544249 100644
--- a/tests-unit/server/routes/internal_routes_test.py
+++ b/tests-unit/server/routes/internal_routes_test.py
@@ -102,7 +102,7 @@ async def test_file_service_initialization():
         # Create a mock instance
         mock_file_service_instance = MagicMock(spec=FileService)
         MockFileService.return_value = mock_file_service_instance
-        internal_routes = InternalRoutes()
+        internal_routes = InternalRoutes(None)
 
         # Check if FileService was initialized with the correct parameters
         MockFileService.assert_called_once_with({