diff --git a/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java b/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java
index 55513cfa76107..172428a54d2fd 100644
--- a/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java
+++ b/azure-mgmt-compute/src/main/java/com/microsoft/azure/management/compute/VirtualMachine.java
@@ -244,7 +244,7 @@ interface Definition extends
             DefinitionStages.WithLinuxRootPasswordOrPublicKey,
             DefinitionStages.WithWindowsAdminUsername,
             DefinitionStages.WithWindowsAdminPassword,
-            DefinitionStages.WithCustomData,
+            DefinitionStages.WithFromImageCreateOptions,
             DefinitionStages.WithLinuxCreate,
             DefinitionStages.WithWindowsCreate,
             DefinitionStages.WithCreate {