diff --git a/tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py b/tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py
index d9ee80347..40182606a 100644
--- a/tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py
+++ b/tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py
@@ -1242,7 +1242,7 @@ def default_partition_fn(keys, shard_num):
   scope = variable_scope.get_variable_scope()
   scope_store = variable_scope._get_default_variable_store()
   full_name = scope.name + "/" + name if scope.name else name
-  if full_name in scope_store._vars:
+  if context.executing_eagerly() and full_name in scope_store._vars:
     if scope.reuse is False:
       err_msg = ("Variable %s already exists, disallowed."
                  " Did you mean to set reuse=True or "