diff --git a/merlin/dataloader/ops/embeddings/embedding_op.py b/merlin/dataloader/ops/embeddings/embedding_op.py
index 8dc496a7..5941882a 100644
--- a/merlin/dataloader/ops/embeddings/embedding_op.py
+++ b/merlin/dataloader/ops/embeddings/embedding_op.py
@@ -108,8 +108,7 @@ def compute_output_schema(
                 name=self.embedding_name,
                 tags=[Tags.CONTINUOUS],
                 dtype=self._get_dtype(self.embeddings),
-                is_list=True,
-                is_ragged=False,
+                properties={"is_list": True, "is_ragged": False},
             )
         )
 
@@ -191,9 +190,11 @@ def compute_output_schema(
                 name=self.embedding_name,
                 tags=[Tags.CONTINUOUS],
                 dtype=self.embeddings.dtype,
-                is_list=True,
-                is_ragged=False,
-                properties={"value_count": {"min": embedding_dim, "max": embedding_dim}},
+                properties={
+                    "is_list": True,
+                    "is_ragged": False,
+                    "value_count": {"min": embedding_dim, "max": embedding_dim},
+                },
             )
         )