diff --git a/src/charm.py b/src/charm.py
index c56f9c6..d2dd510 100755
--- a/src/charm.py
+++ b/src/charm.py
@@ -5,14 +5,14 @@
 import logging
 
 import lightkube
-from charmed_kubeflow_chisme.components.kubernetes_component import KubernetesComponent
 from charmed_kubeflow_chisme.components import (
     CharmReconciler,
     LazyContainerFileTemplate,
     LeadershipGateComponent,
 )
-from charms.mlops_libs.v0.k8s_service_info import KubernetesServiceInfoProvider
+from charmed_kubeflow_chisme.components.kubernetes_component import KubernetesComponent
 from charmed_kubeflow_chisme.kubernetes import create_charm_default_labels
+from charms.mlops_libs.v0.k8s_service_info import KubernetesServiceInfoProvider
 from lightkube.resources.core_v1 import Service
 from ops.charm import CharmBase
 from ops.main import main
diff --git a/tests/unit/test_operator.py b/tests/unit/test_operator.py
index 2141429..f50076b 100644
--- a/tests/unit/test_operator.py
+++ b/tests/unit/test_operator.py
@@ -7,7 +7,7 @@
 from ops.model import ActiveStatus, WaitingStatus
 from ops.testing import Harness
 
-from charm import Operator, GRPC_SVC_NAME, RELATION_NAME
+from charm import GRPC_SVC_NAME, RELATION_NAME, Operator
 
 CONTAINER_NAME = "mlmd-grpc-server"
 MOCK_GRPC_DATA = {"service": "service-name", "port": "1234"}