diff --git a/src/lava/magma/compiler/channel_map.py b/src/lava/magma/compiler/channel_map.py
index eaa129619..4624bc386 100644
--- a/src/lava/magma/compiler/channel_map.py
+++ b/src/lava/magma/compiler/channel_map.py
@@ -96,8 +96,8 @@ def from_proc_groups(self,
         return channel_map
 
     @classmethod
-    def _get_port_pairs_from_proc_groups(
-            self, proc_groups: ty.List[ProcGroup]):
+    def _get_port_pairs_from_proc_groups(self,
+                                         proc_groups: ty.List[ProcGroup]):
         """Loop over processes connectivity and get all connected port pairs."""
         processes = list(itertools.chain.from_iterable(proc_groups))
         port_pairs = []
@@ -117,8 +117,9 @@ def _is_leaf_process_port(dst_port, processes):
         dst_process = dst_port.process
         return True if dst_process in processes else False
 
-    def set_port_initializer(
-        self, port: AbstractPort, port_initializer: PortInitializer
+    def set_port_initializer(self,
+                             port: AbstractPort,
+                             port_initializer: PortInitializer
     ):
         if port in self._initializers_lookup.keys():
             raise AssertionError(
diff --git a/tests/lava/magma/compiler/test_compiler.py b/tests/lava/magma/compiler/test_compiler.py
index 01275e729..0d4e6d07f 100644
--- a/tests/lava/magma/compiler/test_compiler.py
+++ b/tests/lava/magma/compiler/test_compiler.py
@@ -170,7 +170,7 @@ def create_mock_proc_groups() -> ty.List[ProcGroup]:
     py_pg2_p1 = Mock(spec_set=AbstractProcess())
     py_pg2_p2 = Mock(spec_set=AbstractProcess())
 
-    py_pg1_p1.configure_mock(name='py_pg1_p1') 
+    py_pg1_p1.configure_mock(name='py_pg1_p1')
     py_pg2_p1.configure_mock(name='py_pg2_p1')
     py_pg2_p2.configure_mock(name='py_pg2_p2')