diff --git a/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java b/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java
index d162aca1717..229ea31a34b 100644
--- a/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java
+++ b/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java
@@ -387,7 +387,10 @@ public Object invoke(Invocation invocation) {
 
         errorNotification.reset();
         final ControlledUpdateGraph updateGraph = ExecutionContext.getContext().getUpdateGraph().cast();
-        updateGraph.runWithinUnitTestCycle(SUT::refresh);
+        updateGraph.runWithinUnitTestCycle(() -> {
+            SUT.refresh();
+            updateGraph.markSourcesRefreshedForUnitTests();
+        }, false);
         assertIsSatisfied();
         errorNotification.assertInvoked();