diff --git a/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala b/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala
index 00b683c67..9d4451a2f 100644
--- a/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala
+++ b/codepropertygraph/src/main/scala/io/shiftleft/utils/TimeMetric.scala
@@ -254,7 +254,7 @@ object TimeMetric {
     stage: Option[String] = None
   ): (Double, Double, Double, Option[(Double, Double)]) = {
     stage match {
-      case Some(st) =>
+      case Some(st) if stagePerformance.contains(st) =>
         val numberProcessor     = stagePerformance(st)
         val (avgCpu, avgMemory) = numberProcessor.getAverage
         val (maxCpu, maxMemory) = numberProcessor.getMax