diff --git a/engine/table/src/test/java/io/deephaven/engine/table/impl/TestListenerFailure.java b/engine/table/src/test/java/io/deephaven/engine/table/impl/TestListenerFailure.java
index 35299cfbe69..09fcab567b1 100644
--- a/engine/table/src/test/java/io/deephaven/engine/table/impl/TestListenerFailure.java
+++ b/engine/table/src/test/java/io/deephaven/engine/table/impl/TestListenerFailure.java
@@ -4,6 +4,7 @@
 package io.deephaven.engine.table.impl;
 
 import io.deephaven.engine.context.ExecutionContext;
+import io.deephaven.engine.exceptions.TableAlreadyFailedException;
 import io.deephaven.engine.table.Table;
 import io.deephaven.engine.testutil.ControlledUpdateGraph;
 import io.deephaven.engine.testutil.TstUtils;
@@ -48,8 +49,8 @@ public void testListenerFailure() {
         try {
             updated.addUpdateListener(new ErrorListener(updated));
             TestCase.fail("Should not be allowed to listen to failed table");
-        } catch (IllegalStateException ise) {
-            assertEquals("Can not listen to failed table QueryTable", ise.getMessage());
+        } catch (TableAlreadyFailedException tafe) {
+            assertEquals("Can not listen to failed table QueryTable", tafe.getMessage());
         }
 
         try {
@@ -59,8 +60,8 @@ public void testListenerFailure() {
                         public void onUpdate(RowSet added, RowSet removed, RowSet modified) {}
                     }, false);
             TestCase.fail("Should not be allowed to listen to failed table");
-        } catch (IllegalStateException ise) {
-            assertEquals("Can not listen to failed table QueryTable", ise.getMessage());
+        } catch (TableAlreadyFailedException tafe) {
+            assertEquals("Can not listen to failed table QueryTable", tafe.getMessage());
         }
     }
 
diff --git a/engine/table/src/test/java/io/deephaven/engine/util/systemicmarking/TestSystemicObjectMarking.java b/engine/table/src/test/java/io/deephaven/engine/util/systemicmarking/TestSystemicObjectMarking.java
index 2748f0d7612..4b34c5cd813 100644
--- a/engine/table/src/test/java/io/deephaven/engine/util/systemicmarking/TestSystemicObjectMarking.java
+++ b/engine/table/src/test/java/io/deephaven/engine/util/systemicmarking/TestSystemicObjectMarking.java
@@ -4,6 +4,7 @@
 package io.deephaven.engine.util.systemicmarking;
 
 import io.deephaven.engine.context.ExecutionContext;
+import io.deephaven.engine.exceptions.TableAlreadyFailedException;
 import io.deephaven.engine.table.Table;
 import io.deephaven.engine.table.impl.ErrorListener;
 import io.deephaven.engine.table.impl.QueryTable;
@@ -60,8 +61,8 @@ public void testSystemicObjectMarking() {
         try {
             updated2.addUpdateListener(new ErrorListener(updated2));
             TestCase.fail("Should not be allowed to listen to failed table");
-        } catch (IllegalStateException ise) {
-            assertEquals("Can not listen to failed table QueryTable", ise.getMessage());
+        } catch (TableAlreadyFailedException tafe) {
+            assertEquals("Can not listen to failed table QueryTable", tafe.getMessage());
         }
 
         final ErrorListener errorListener = new ErrorListener(updated);