diff --git a/src/mlstacks/terraform/aws-modular/output_file.tf b/src/mlstacks/terraform/aws-modular/output_file.tf
index 9c628896..381da617 100644
--- a/src/mlstacks/terraform/aws-modular/output_file.tf
+++ b/src/mlstacks/terraform/aws-modular/output_file.tf
@@ -29,7 +29,7 @@ resource "local_file" "stack_file" {
 %{endif}
 
       orchestrator:
-%{if var.enable_orchestrator_kubeflow}      
+%{if var.enable_orchestrator_kubeflow}
         id: ${uuid()}
         flavor: kubeflow
         name: eks_kubeflow_orchestrator
@@ -43,7 +43,7 @@ resource "local_file" "stack_file" {
 %{else}
 %{if var.enable_orchestrator_skypilot}
         id: ${uuid()}
-        flavor: vm-aws
+        flavor: vm_aws
         name: aws_skypilot_orchestrator
         configuration: {}
 %{else}
diff --git a/src/mlstacks/terraform/gcp-modular/output_file.tf b/src/mlstacks/terraform/gcp-modular/output_file.tf
index 086f7bb5..2a86d029 100644
--- a/src/mlstacks/terraform/gcp-modular/output_file.tf
+++ b/src/mlstacks/terraform/gcp-modular/output_file.tf
@@ -49,7 +49,7 @@ resource "local_file" "stack_file" {
 %{else}
 %{if var.enable_orchestrator_skypilot}
         id: ${uuid()}
-        flavor: vm-gcp
+        flavor: vm_gcp
         name: gcp_skypilot_orchestrator
         configuration: {"project_id": "${var.project_id}"}
 %{else}