diff --git a/ui/scripts/ui-custom/affinity.js b/ui/scripts/ui-custom/affinity.js
index 6c237ed92eb..1f51b52ec59 100644
--- a/ui/scripts/ui-custom/affinity.js
+++ b/ui/scripts/ui-custom/affinity.js
@@ -103,6 +103,7 @@
                   response: {
                     success: function(args) {
                       complete({
+                        _custom: args._custom,
                         $item: $instanceRow
                       });
                     },