diff --git a/b_asic/signal_flow_graph.py b/b_asic/signal_flow_graph.py
index fdcfa9fd1475d154c8bf9603fc9d00ea60427be7..d51f13b4209fd1d5fc87e369b2f23dc8bf69301b 100644
--- a/b_asic/signal_flow_graph.py
+++ b/b_asic/signal_flow_graph.py
@@ -627,7 +627,10 @@ class SFG(AbstractOperation):
             for port in ports:
                 for signal in port.signals:
                     pg.edge(port.operation.graph_id + '.' + str(port.index), signal.destination.operation.graph_id)
+                    pg.node(signal.destination.operation.graph_id, shape = 'square')
                 pg.edge(port.operation.graph_id, port.operation.graph_id + '.' + str(port.index))
+                pg.node(port.operation.graph_id, shape = 'square')
+
         pg.view()  
 
     def print_precedence_graph(self) -> None: