From 55ad6639f2b209d590454e2d595ea8bdffca9fc5 Mon Sep 17 00:00:00 2001
From: Derek Christ <dchrist@rhrk.uni-kl.de>
Date: Tue, 25 Jan 2022 15:31:17 +0100
Subject: [PATCH] Fix a bug where more transactions were simulated than
 specified

---
 DRAMSys/simulator/TrafficGenerator.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/DRAMSys/simulator/TrafficGenerator.cpp b/DRAMSys/simulator/TrafficGenerator.cpp
index 539b0f6f..5d15235d 100644
--- a/DRAMSys/simulator/TrafficGenerator.cpp
+++ b/DRAMSys/simulator/TrafficGenerator.cpp
@@ -277,6 +277,12 @@ void TrafficGenerator::transitionToNextState()
 
 void TrafficGenerator::prepareNextPayload()
 {
+    if (transactionsSent >= maxTransactions)
+    {
+        finished = true;
+        return;
+    }
+
     if (startEvent && transactionsSent == 0)
         wait(*startEvent);