diff --git a/b_asic/schedule.py b/b_asic/schedule.py
index e5d5f96aba727e77d309cd25f47d2473da1b364a..6d5a8364ca0f1df40b5e3e9459a97220d187c0ea 100644
--- a/b_asic/schedule.py
+++ b/b_asic/schedule.py
@@ -113,7 +113,6 @@ class Schedule:
         if not isinstance(sfg, SFG):
             raise TypeError("An SFG must be provided")
 
-        self._original_sfg = sfg()  # Make a copy
         self._sfg = sfg
         self._start_times = {}
         self._laps = defaultdict(_laps_default)
@@ -390,13 +389,14 @@ class Schedule:
         operation_id : GraphID
             The GraphID of the operation to swap.
         """
-        self._original_sfg.swap_io_of_operation(operation_id)
         self._sfg.swap_io_of_operation(operation_id)
 
     @property
     def sfg(self) -> SFG:
-        """The SFG of the current schedule."""
-        return self._original_sfg
+        """The SFG corresponding to the current schedule."""
+        reconstructed_sfg = self._reintroduce_delays()
+        simplified_sfg = reconstructed_sfg.simplify_delay_element_placement()
+        return simplified_sfg
 
     @property
     def start_times(self) -> Dict[GraphID, int]:
@@ -529,7 +529,6 @@ class Schedule:
             The execution time of the operation.
         """
         self._sfg.set_execution_time_of_type(type_name, execution_time)
-        self._original_sfg.set_execution_time_of_type(type_name, execution_time)
 
     def move_y_location(
         self, graph_id: GraphID, new_y: int, insert: bool = False
@@ -761,6 +760,7 @@ class Schedule:
             delay_list = self._sfg.find_by_type_name(Delay.type_name())
 
     def _reintroduce_delays(self) -> SFG:
+        """Reintroduce delay elements to each signal according to the ``_laps`` variable."""
         reconstructed_sfg = self._sfg()
         for signal_id,lap in self._laps.items():
             for delays in range(lap):
diff --git a/b_asic/scheduler_gui/main_window.py b/b_asic/scheduler_gui/main_window.py
index c1740e5814762b19764cfdc41186f0debea6d1dd..437dc0098f5f66ab83d2bbfa7490561392d93b62 100644
--- a/b_asic/scheduler_gui/main_window.py
+++ b/b_asic/scheduler_gui/main_window.py
@@ -439,7 +439,6 @@ class ScheduleMainWindow(QMainWindow, Ui_MainWindow):
             self.save_as()
             return
         self._schedule._sfg._graph_id_generator = None
-        self._schedule._original_sfg._graph_id_generator = None
         with open(self._file_name, 'wb') as f:
             pickle.dump(self._schedule, f)
         self._add_recent_file(self._file_name)
@@ -462,7 +461,6 @@ class ScheduleMainWindow(QMainWindow, Ui_MainWindow):
             filename += '.bsc'
         self._file_name = filename
         self._schedule._sfg._graph_id_generator = None
-        self._schedule._original_sfg._graph_id_generator = None
         with open(self._file_name, 'wb') as f:
             pickle.dump(self._schedule, f)
         self._add_recent_file(self._file_name)
diff --git a/b_asic/signal_flow_graph.py b/b_asic/signal_flow_graph.py
index 19ef0a6028f60b84834b0de4f45b77137bd13e34..7139806dc9d39a29846d4bb751972a210271c192 100644
--- a/b_asic/signal_flow_graph.py
+++ b/b_asic/signal_flow_graph.py
@@ -737,6 +737,13 @@ class SFG(AbstractOperation):
         return sfg_copy()
 
     def simplify_delay_element_placement(self) -> "SFG":
+        """ Simplify an SFG by removing some redundant delay elements.
+            For example two signals originating from the same starting point, each
+            connected to a delay element will combine into a single delay element.
+
+            Returns a copy of the simplified SFG.
+        """
+
         sfg_copy = self()
         for delay_element in sfg_copy.find_by_type_name(Delay.type_name()):
             neighboring_delays = []
diff --git a/test/test_schedule.py b/test/test_schedule.py
index 6990eb891ea511dc0a068a0bcb236355fc60fc4c..aba52d2f1d39e52a490fb6016c8d752b2b6aad30 100644
--- a/test/test_schedule.py
+++ b/test/test_schedule.py
@@ -520,6 +520,23 @@ class TestRescheduling:
         assert schedule._start_times["add0"] == 0
         assert schedule._start_times["out0"] == 2
 
+    def test_reintroduce_delays(self, precedence_sfg_delays, sfg_direct_form_iir_lp_filter):
+        precedence_sfg_delays.set_latency_of_type(Addition.type_name(), 1)
+        precedence_sfg_delays.set_latency_of_type(ConstantMultiplication.type_name(), 3)
+        sfg_direct_form_iir_lp_filter.set_latency_of_type(Addition.type_name(), 1)
+        sfg_direct_form_iir_lp_filter.set_latency_of_type(ConstantMultiplication.type_name(), 3)
+
+        schedule = Schedule(precedence_sfg_delays, algorithm="ASAP")
+        sfg = schedule.sfg
+        assert precedence_sfg_delays.evaluate(5) == sfg.evaluate(5)
+
+        schedule = Schedule(sfg_direct_form_iir_lp_filter, algorithm="ASAP")
+        sfg = schedule.sfg
+        assert sfg_direct_form_iir_lp_filter.evaluate(5) == sfg.evaluate(5)
+
+
+
+
 
 class TestTimeResolution:
     def test_increase_time_resolution(