diff --git a/packages/core/src/api/config/graphql-custom-fields.ts b/packages/core/src/api/config/graphql-custom-fields.ts
index ec771bd22e..df4f44c00b 100644
--- a/packages/core/src/api/config/graphql-custom-fields.ts
+++ b/packages/core/src/api/config/graphql-custom-fields.ts
@@ -198,6 +198,13 @@ export function addGraphQLCustomFields(
                 }
             `;
         }
+        if (schema.getType('UpdateOrderAddressInput')) {
+            customFieldTypeDefs += `
+                extend input UpdateOrderAddressInput {
+                    customFields: UpdateAddressCustomFieldsInput
+                }
+            `;
+        }
     } else {
         if (schema.getType('OrderAddress')) {
             customFieldTypeDefs += `