diff --git a/src/main/java/io/debezium/connector/vitess/VitessSnapshotChangeEventSource.java b/src/main/java/io/debezium/connector/vitess/VitessSnapshotChangeEventSource.java
index 4824eed1..9bdb77b4 100644
--- a/src/main/java/io/debezium/connector/vitess/VitessSnapshotChangeEventSource.java
+++ b/src/main/java/io/debezium/connector/vitess/VitessSnapshotChangeEventSource.java
@@ -87,8 +87,8 @@ public SnapshottingTask getSnapshottingTask(VitessPartition partition, VitessOff
     }
 
     @Override
-    protected SnapshotContext<VitessPartition, VitessOffsetContext> prepare(VitessPartition partition) {
-        return new RelationalSnapshotContext<>(partition, "");
+    protected SnapshotContext<VitessPartition, VitessOffsetContext> prepare(VitessPartition partition, boolean onDemand) {
+        return new RelationalSnapshotContext<>(partition, "", onDemand);
     }
 
     @Override