From 63ea2e290e0f6fecf5d76901b434842596c25a99 Mon Sep 17 00:00:00 2001
From: ableegoldman <sophie@confluent.io>
Date: Wed, 23 Oct 2019 14:00:22 -0700
Subject: [PATCH] fix bug

---
 .../org/apache/kafka/streams/tests/StreamsUpgradeTest.java    | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/streams/src/test/java/org/apache/kafka/streams/tests/StreamsUpgradeTest.java b/streams/src/test/java/org/apache/kafka/streams/tests/StreamsUpgradeTest.java
index 185fa7c3bb14e..318154889ca60 100644
--- a/streams/src/test/java/org/apache/kafka/streams/tests/StreamsUpgradeTest.java
+++ b/streams/src/test/java/org/apache/kafka/streams/tests/StreamsUpgradeTest.java
@@ -240,7 +240,9 @@ public GroupAssignment assign(final Cluster metadata, final GroupSubscription gr
                                 info.prevTasks(),
                                 info.standbyTasks(),
                                 info.userEndPoint())
-                                .encode()));
+                                .encode(),
+                            subscription.ownedPartitions()
+                        ));
                 }
                 assignment = super.assign(metadata, new GroupSubscription(downgradedSubscriptions)).groupAssignment();
                 bumpUsedVersion = true;