diff --git a/runners/flink/job-server/flink_job_server.gradle b/runners/flink/job-server/flink_job_server.gradle
index b85f8fc98aaa..335439cbda91 100644
--- a/runners/flink/job-server/flink_job_server.gradle
+++ b/runners/flink/job-server/flink_job_server.gradle
@@ -193,7 +193,7 @@ def portableValidatesRunnerTask(String name, String mode, boolean checkpointing,
           excludeCategories 'org.apache.beam.sdk.testing.UsesTriggeredSideInputs'
           return
         }
-        if (mode == "batch") {
+        if (mode == "batch" || mode == "batch-dataset") {
             excludeCategories 'org.apache.beam.sdk.testing.UsesTriggeredSideInputs'
         }
         excludeCategories 'org.apache.beam.sdk.testing.UsesUnboundedSplittableParDo'