diff --git a/quickstart/301-machine-learning-hub-spoke-secure/azure-firewall.tf b/quickstart/301-machine-learning-hub-spoke-secure/azure-firewall.tf
index 6293ab1cb..dd19174d5 100644
--- a/quickstart/301-machine-learning-hub-spoke-secure/azure-firewall.tf
+++ b/quickstart/301-machine-learning-hub-spoke-secure/azure-firewall.tf
@@ -48,6 +48,8 @@ resource "azurerm_firewall" "azure_firewall_instance" {
   location            = azurerm_resource_group.default.location
   resource_group_name = azurerm_resource_group.hub_rg.name
   firewall_policy_id  = azurerm_firewall_policy.base_policy.id
+  sku_name            = "AZFW_VNet"
+  sku_tier            = "Standard"
 
   ip_configuration {
     name                 = "configuration"
@@ -487,4 +489,4 @@ resource "azurerm_firewall_policy_rule_collection_group" "azure_firewall_rules_c
     azurerm_ip_group.ip_group_hub,
     azurerm_ip_group.ip_group_spoke
   ]
-}
\ No newline at end of file
+}
diff --git a/quickstart/301-machine-learning-hub-spoke-secure/main.tf b/quickstart/301-machine-learning-hub-spoke-secure/main.tf
index 074cec954..cdafd5ea3 100644
--- a/quickstart/301-machine-learning-hub-spoke-secure/main.tf
+++ b/quickstart/301-machine-learning-hub-spoke-secure/main.tf
@@ -4,11 +4,11 @@ terraform {
   required_providers {
     azurerm = {
       source  = "hashicorp/azurerm"
-      version = "=2.78.0"
+      version = ">=2.78.0"
     }
 
     azureml = {
-      source = "registry.terraform.io/Telemaco019/azureml"
+      source = "registry.terraform.io/orobix/azureml"
     }
   }
 }
@@ -29,4 +29,4 @@ resource "azurerm_resource_group" "hub_rg" {
   name     = "rg-hub-${var.name}-${var.environment}"
   location = var.location
 
-}
\ No newline at end of file
+}
diff --git a/quickstart/301-machine-learning-hub-spoke-secure/variables.tf b/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
index 9618d98c6..d2969abaa 100644
--- a/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
+++ b/quickstart/301-machine-learning-hub-spoke-secure/variables.tf
@@ -1,6 +1,7 @@
 variable "name" {
   type        = string
   description = "Name of the deployment"
+  default     = "exampleml"
 }
 
 variable "environment" {
@@ -89,5 +90,6 @@ variable "dsvm_admin_username" {
 variable "dsvm_host_password" {
   type        = string
   description = "Password for the admin username of the Data Science VM"
+  default     = "ChangeMe123!"
   sensitive   = true
-}
\ No newline at end of file
+}