diff --git a/llvm/projects/soc_simulator/src/driver.py b/llvm/projects/soc_simulator/src/driver.py
index 9d39f1394f6faa55f3eb00bb78ddf13747e3738a..dbf2651bd3a9512c46d9e0a549c61290ad913ab0 100644
--- a/llvm/projects/soc_simulator/src/driver.py
+++ b/llvm/projects/soc_simulator/src/driver.py
@@ -264,6 +264,7 @@ class Driver:
 
                     self.__aggregate_results[Driver.results_time_key][self.__config_count] += total_time
                     self.__aggregate_results[Driver.results_energy_key][self.__config_count] += total_energy 
+                prev_layer = curr_layer
 
             self.__config_count += 1
             print("\n")