diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/job_definition_schema.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/job_definition_schema.py
index 0e2a467bdac5..cfe60ba66d49 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/job_definition_schema.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/job_definition_schema.py
@@ -9,7 +9,14 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._schema import ArmVersionedStr, PatchedSchemaMeta, StringTransformedEnum, UnionField, ArmStr
+from azure.ai.ml._schema import (
+    ArmVersionedStr,
+    PatchedSchemaMeta,
+    StringTransformedEnum,
+    UnionField,
+    ArmStr,
+    RegistryStr,
+)
 from azure.ai.ml._schema.pipeline.pipeline_component import PipelineComponentFileRefField
 from azure.ai.ml.constants._common import AzureMLResourceType
 from azure.ai.ml.constants._job.job import JobType
@@ -27,6 +34,7 @@ class JobDefinitionSchema(metaclass=PatchedSchemaMeta):
     )
     component = UnionField(
         [
+            RegistryStr(azureml_type=AzureMLResourceType.COMPONENT),
             ArmVersionedStr(azureml_type=AzureMLResourceType.COMPONENT, allow_default_version=True),
             PipelineComponentFileRefField(),
         ]