From 252e599a6cffaeeba1d82ae5a03639d85e177810 Mon Sep 17 00:00:00 2001
From: Brian Federle <brian.federle@citrix.com>
Date: Thu, 28 Jun 2012 14:20:17 -0700
Subject: [PATCH] Pass _custom correctly to VPC tier add VM action

---
 ui/scripts/ui-custom/vpc.js | 4 +++-
 ui/scripts/vpc.js           | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ui/scripts/ui-custom/vpc.js b/ui/scripts/ui-custom/vpc.js
index 0545748e0cc..851e5a827b2 100644
--- a/ui/scripts/ui-custom/vpc.js
+++ b/ui/scripts/ui-custom/vpc.js
@@ -207,12 +207,14 @@
 
     var success = function(args) {
       var remove = args ? args.remove : false;
+      var _custom = args ? args._custom : {};
 
       cloudStack.ui.notifications.add(
         // Notification
         {
           desc: label,
-          poll: notification.poll
+          poll: notification.poll,
+          _custom: _custom
         },
 
         // Success
diff --git a/ui/scripts/vpc.js b/ui/scripts/vpc.js
index bc3cf70d769..7e04c7005ac 100644
--- a/ui/scripts/vpc.js
+++ b/ui/scripts/vpc.js
@@ -427,7 +427,7 @@
             cloudStack.instanceWizard
           ),
           notification: {
-            poll: function(args) { args.complete(); }
+            poll: pollAsyncJobResult
           }
         },
         acl: {