diff --git a/rocket/src/main/scala/nbdcache.scala b/rocket/src/main/scala/nbdcache.scala
index 6b684f64..40cac1c8 100644
--- a/rocket/src/main/scala/nbdcache.scala
+++ b/rocket/src/main/scala/nbdcache.scala
@@ -821,7 +821,8 @@ class HellaCache(implicit p: Parameters) extends L1HellaCacheModule()(p) {
 
   assert (!(Reg(next=
     (io.cpu.xcpt.ma.ld || io.cpu.xcpt.ma.st || io.cpu.xcpt.pf.ld || io.cpu.xcpt.pf.st)) &&
-    io.cpu.resp.valid), "DCache exception occurred - cache response not killed.")
+    s2_valid_masked),
+      "DCache exception occurred - cache response not killed.")
 
   // tags
   def onReset = L1Metadata(UInt(0), ClientMetadata.onReset)