diff --git a/smoke-test/tests/test_stateful_ingestion.py b/smoke-test/tests/test_stateful_ingestion.py
index be3a8f9aa9131..b6861b18d2a57 100644
--- a/smoke-test/tests/test_stateful_ingestion.py
+++ b/smoke-test/tests/test_stateful_ingestion.py
@@ -114,7 +114,7 @@ def get_current_checkpoint_from_pipeline(
     state1 = cast(BaseSQLAlchemyCheckpointState, checkpoint1.state)
     state2 = cast(BaseSQLAlchemyCheckpointState, checkpoint2.state)
     difference_urns = list(
-        state1.get_urns_not_in(type="table", other_checkpoint_state=state2)
+        state1.get_urns_not_in(type="*", other_checkpoint_state=state2)
     )
     assert len(difference_urns) == 1
     assert (