diff --git a/hapi-fhir-sql-migrate/src/main/java/ca/uhn/fhir/jpa/migrate/tasks/api/Builder.java b/hapi-fhir-sql-migrate/src/main/java/ca/uhn/fhir/jpa/migrate/tasks/api/Builder.java
index 5b9822475b11..ad830f51f11e 100644
--- a/hapi-fhir-sql-migrate/src/main/java/ca/uhn/fhir/jpa/migrate/tasks/api/Builder.java
+++ b/hapi-fhir-sql-migrate/src/main/java/ca/uhn/fhir/jpa/migrate/tasks/api/Builder.java
@@ -163,9 +163,10 @@ public void addIdGenerator(String theVersion, String theGeneratorName) {
 		addTask(task);
 	}
 
-	public void dropIdGenerator(String theVersion, String theIdGeneratorName) {
+	public DropIdGeneratorTask dropIdGenerator(String theVersion, String theIdGeneratorName) {
 		DropIdGeneratorTask task = new DropIdGeneratorTask(myRelease, theVersion, theIdGeneratorName);
 		addTask(task);
+		return task;
 	}
 
 	public void addNop(String theVersion) {