diff --git a/mlos_bench/mlos_bench/config/services/remote/azure/scripts/setup-rg/rg-template.json b/mlos_bench/mlos_bench/config/services/remote/azure/scripts/setup-rg/rg-template.json
index a9676ebf078..8fbe12e518c 100644
--- a/mlos_bench/mlos_bench/config/services/remote/azure/scripts/setup-rg/rg-template.json
+++ b/mlos_bench/mlos_bench/config/services/remote/azure/scripts/setup-rg/rg-template.json
@@ -53,7 +53,7 @@
         },
         "storageSku": {
             "type": "string",
-            "defaultValue": "Premium_LRS",
+            "defaultValue": "Standard_LRS",
             "allowedValues": [
                 "Premium_LRS",
                 "Premium_ZRS",
@@ -64,6 +64,14 @@
                 "Standard_RAGZRS",
                 "Standard_ZRS"
             ]
+        },
+        "storageKind": {
+            "type": "string",
+            "defaultValue": "StorageV2",
+            "allowedValues": [
+                "FileStorage",
+                "StorageV2"
+            ]
         }
     },
     "functions": [],
@@ -299,7 +307,7 @@
         "dependsOn": [
             "[resourceId('Microsoft.Network/virtualNetworks', variables('vnetName'))]"
         ],
-        "kind": "StorageV2",
+        "kind": "[parameters('storageKind')]",
         "sku": {
             "name": "[parameters('storageSku')]"
         },
@@ -312,7 +320,11 @@
                         "action": "Allow"
                     }
                 ],
-                "ipRules": [],
+                "ipRules": [
+                    {
+                        "value": "[parameters('vmSshSourceAddressPrefix')]"
+                    }
+                ],
                 "defaultAction": "Deny"
             }
         }