diff --git a/src/telemetry_frontend/tests/test_unitary.py b/src/telemetry_frontend/tests/test_unitary.py
index c52d689211a00dda7b717c0283b766cc043f75c4..4f37514dc874531a39499fc509bea56e828a7b17 100644
--- a/src/telemetry_frontend/tests/test_unitary.py
+++ b/src/telemetry_frontend/tests/test_unitary.py
@@ -13,19 +13,22 @@
 # limitations under the License.
 
 import os
+import pytest
 import logging
+from typing import Union
 from common.Constants import ServiceNameEnum
 from common.proto.telemetry_frontend_pb2 import CollectorId
+from common.proto.context_pb2_grpc import add_ContextServiceServicer_to_server
 from context.client.ContextClient import ContextClient
-
+from common.tools.service.GenericGrpcService import GenericGrpcService
+from common.tests.MockServicerImpl_Context import MockServicerImpl_Context
+from common.Settings import ( 
+    get_service_port_grpc, get_env_var_name, ENVVAR_SUFIX_SERVICE_HOST, ENVVAR_SUFIX_SERVICE_PORT_GRPC)
 
 from telemetry_frontend.client.TelemetryFrontendClient import TelemetryFrontendClient
 from telemetry_frontend.service.TelemetryFrontendService import TelemetryFrontendService
 from telemetry_frontend.tests.Messages import create_collector_request
 
-from common.Settings import ( 
-    get_service_port_grpc, get_env_var_name, ENVVAR_SUFIX_SERVICE_HOST, ENVVAR_SUFIX_SERVICE_PORT_GRPC)
-
 from device.client.DeviceClient import DeviceClient
 from device.service.DeviceService import DeviceService
 from device.service.driver_api.DriverFactory import DriverFactory