Small update to the snapshot (#330)

1 files changed, 1 insertions, 1 deletions

diff --git a/opendc-simulator/opendc-simulator-compute/src/main/java/org/opendc/simulator/compute/workload/trace/SimTraceWorkload.java b/opendc-simulator/opendc-simulator-compute/src/main/java/org/opendc/simulator/compute/workload/trace/SimTraceWorkload.java
index 9e7f170c..d5635439 100644
--- a/opendc-simulator/opendc-simulator-compute/src/main/java/org/opendc/simulator/compute/workload/trace/SimTraceWorkload.java
+++ b/opendc-simulator/opendc-simulator-compute/src/main/java/org/opendc/simulator/compute/workload/trace/SimTraceWorkload.java
@@ -204,7 +204,7 @@ public class SimTraceWorkload extends SimWorkload implements FlowConsumer {
                 this.scalingPolicy.getRemainingDuration(this.cpuFreqDemand, this.cpuFreqDemand, this.remainingWork);
 
         // If this is the end of the Task, don't make a snapshot
-        if (remainingTime <= 0 && remainingFragments.isEmpty()) {
+        if (this.currentFragment == null || (remainingTime <= 0 && remainingFragments.isEmpty())) {
             return;
         }