diff --git a/gerrit-server/src/main/java/com/google/gerrit/server/notedb/ChangeUpdate.java b/gerrit-server/src/main/java/com/google/gerrit/server/notedb/ChangeUpdate.java
index 74ca198cdb..357d635206 100644
--- a/gerrit-server/src/main/java/com/google/gerrit/server/notedb/ChangeUpdate.java
+++ b/gerrit-server/src/main/java/com/google/gerrit/server/notedb/ChangeUpdate.java
@@ -182,7 +182,7 @@ public class ChangeUpdate extends VersionedMetaData {
 
   public void setPatchSetId(PatchSet.Id psId) {
     checkArgument(psId == null
-        || psId.getParentKey().equals(getChange().getKey()));
+        || psId.getParentKey().equals(getChange().getId()));
     this.psId = psId;
   }