diff --git a/simplifier.py b/simplifier.py
index d02eeb9..21207b7 100644
--- a/simplifier.py
+++ b/simplifier.py
@@ -50,6 +50,9 @@ def eliminate_forwarders(self):
                         if forwardedBB_in_edge_exists and forwarderBB_in_edge_exists:
                             continue
 
+                        if forwarderBB in nx.get_node_attributes(self.bb_graph,'isEntry').keys():
+                            nx.set_node_attributes(self.bb_graph, 'isEntry', {forwardedBB: True})
+
                         # Remove the edge between forwarder and forwarded
                         self.bb_graph.remove_edge(forwarderBB, forwardedBB)