diff --git a/tensorflow_recommenders/layers/factorized_top_k.py b/tensorflow_recommenders/layers/factorized_top_k.py
index b6a7f45..6159803 100644
--- a/tensorflow_recommenders/layers/factorized_top_k.py
+++ b/tensorflow_recommenders/layers/factorized_top_k.py
@@ -373,7 +373,7 @@ def __init__(self,
     self._num_parallel_calls = num_parallel_calls
     self._sorted = sorted_order
 
-    self._counter = self.add_weight("counter", dtype=tf.int32, trainable=False)
+    self._counter = self.add_weight(name="counter", dtype=tf.int32, trainable=False)
 
   def index_from_dataset(
       self,