diff --git a/gerrit-server/src/main/java/com/google/gerrit/server/change/RebaseChangeEdit.java b/gerrit-server/src/main/java/com/google/gerrit/server/change/RebaseChangeEdit.java
index 9046016046..53ee07fc91 100644
--- a/gerrit-server/src/main/java/com/google/gerrit/server/change/RebaseChangeEdit.java
+++ b/gerrit-server/src/main/java/com/google/gerrit/server/change/RebaseChangeEdit.java
@@ -74,7 +74,7 @@ public class RebaseChangeEdit implements
   }
 
   @Singleton
-  public static class Rebase implements RestModifyView<ChangeResource, PublishDraftPatchSet.Input> {
+  public static class Rebase implements RestModifyView<ChangeResource, Rebase.Input> {
     public static class Input {
     }
 
@@ -92,7 +92,7 @@ public class RebaseChangeEdit implements
     }
 
     @Override
-    public Response<?> apply(ChangeResource rsrc, PublishDraftPatchSet.Input in)
+    public Response<?> apply(ChangeResource rsrc, Rebase.Input in)
         throws AuthException, ResourceConflictException, IOException,
         InvalidChangeOperationException, OrmException {
       Optional<ChangeEdit> edit = editUtil.byChange(rsrc.getChange());