diff --git a/nds/jvm_listener/src/main/scala/com/nvidia/spark/rapids/listener/TaskFailureListener.scala b/nds/jvm_listener/src/main/scala/com/nvidia/spark/rapids/listener/TaskFailureListener.scala
index 791be72..6a894d5 100644
--- a/nds/jvm_listener/src/main/scala/com/nvidia/spark/rapids/listener/TaskFailureListener.scala
+++ b/nds/jvm_listener/src/main/scala/com/nvidia/spark/rapids/listener/TaskFailureListener.scala
@@ -17,7 +17,7 @@
 
 package com.nvidia.spark.rapids.listener
 
-import org.apache.spark.{Success, TaskEndReason}
+import org.apache.spark.{Success, TaskEndReason, TaskFailedReason, TaskKilled}
 import org.apache.spark.scheduler.{SparkListener, SparkListenerTaskEnd}
 import scala.collection.mutable.ListBuffer
 
@@ -29,8 +29,9 @@ import scala.collection.mutable.ListBuffer
 class TaskFailureListener extends SparkListener {
   override def onTaskEnd(taskEnd: SparkListenerTaskEnd) {
     taskEnd.reason match {
-      case Success =>
-      case reason => Manager.notifyAll(reason.toString)
+      case Success => ()
+      case _: TaskKilled => ()
+      case failedReason: TaskFailedReason => Manager.notifyAll(failedReason.toErrorString)
     }
     super.onTaskEnd(taskEnd)
   }