diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java
index 296a53f48e80..fd178f8e7649 100644
--- a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java
+++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java
@@ -22,6 +22,7 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
 import org.apache.beam.sdk.coders.Coder;
 import org.apache.beam.sdk.coders.KvCoder;
 import org.apache.beam.sdk.coders.StringUtf8Coder;
@@ -226,5 +227,19 @@ public long getNum() {
     public String getStr() {
       return this.str;
     }
+
+    @Override
+    public boolean equals(Object o) {
+      if (!(o instanceof Pojo)) {
+        return false;
+      }
+      Pojo pojo = (Pojo) o;
+      return num == pojo.num && Objects.equals(str, pojo.str);
+    }
+
+    @Override
+    public int hashCode() {
+      return Objects.hash(num, str);
+    }
   }
 }