diff --git a/lib/shared/addon/components/machine/driver-azure/component.js b/lib/shared/addon/components/machine/driver-azure/component.js
index c2f7f44ec..d1f39c0e6 100644
--- a/lib/shared/addon/components/machine/driver-azure/component.js
+++ b/lib/shared/addon/components/machine/driver-azure/component.js
@@ -202,5 +202,8 @@ export default Component.extend(Driver, {
 
     return true;
   },
+  doneSaving(neu) {
+    this.sendAction('completed', neu);
+  },
 
 });
diff --git a/lib/shared/addon/components/modal-add-cluster/template.hbs b/lib/shared/addon/components/modal-add-cluster/template.hbs
index 2e4813d60..0ad3ff3e8 100644
--- a/lib/shared/addon/components/modal-add-cluster/template.hbs
+++ b/lib/shared/addon/components/modal-add-cluster/template.hbs
@@ -16,5 +16,5 @@
       save=(action 'launch')
   }}
 {{else if (eq mode 'add')}}
-  {{add-host allowCustom=(not app.isCaas) model=addHostModel.model driver=driver cluster=cluster hostId=hostId inModal=true completed=(action 'completed') goBack=(action 'goBack') cancel=(action 'goBack')}}
+  {{add-host allowCustom=(not app.isCaas) model=addHostModel.model driver=driver cluster=cluster hostId=hostId completed=(action 'completed') goBack=(action 'goBack') cancel=(action 'goBack')}}
 {{/if}}