diff --git a/faust/__init__.py b/faust/__init__.py
index 67b57dc79..2c2e15d1e 100644
--- a/faust/__init__.py
+++ b/faust/__init__.py
@@ -23,7 +23,7 @@
 import typing
 from typing import Any, Mapping, NamedTuple, Optional, Sequence, Tuple
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 __author__ = "Robinhood Markets, Inc."
 __contact__ = "schrohm@gmail.com, vpatki@wayfair.com"
 __homepage__ = "https://github.com/faust-streaming/faust"
diff --git a/faust/tables/base.py b/faust/tables/base.py
index 7423aa391..26608ef14 100644
--- a/faust/tables/base.py
+++ b/faust/tables/base.py
@@ -568,9 +568,7 @@ async def on_rebalance(
         generation_id: int = 0,
     ) -> None:
         """Call when cluster is rebalancing."""
-        await self.data.on_rebalance(
-            self, assigned, revoked, newly_assigned, generation_id
-        )
+        await self.data.on_rebalance(assigned, revoked, newly_assigned, generation_id)
 
     async def on_recovery_completed(
         self, active_tps: Set[TP], standby_tps: Set[TP]
diff --git a/tests/unit/tables/test_base.py b/tests/unit/tables/test_base.py
index 5543d34cc..8b5733f2a 100644
--- a/tests/unit/tables/test_base.py
+++ b/tests/unit/tables/test_base.py
@@ -559,7 +559,7 @@ async def test_on_rebalance(self, *, table):
         generation_id = 1
         await table.on_rebalance({TP1}, set(), set(), generation_id)
         table._data.on_rebalance.assert_called_once_with(
-            table, {TP1}, set(), set(), generation_id
+            {TP1}, set(), set(), generation_id
         )
 
     @pytest.mark.asyncio