diff --git a/addons/api/mirage/factories/host.js b/addons/api/mirage/factories/host.js
index 895b44468..60300cba4 100644
--- a/addons/api/mirage/factories/host.js
+++ b/addons/api/mirage/factories/host.js
@@ -10,4 +10,9 @@ export default factory.extend({
       'delete',
     ],
   id: (i) => `host-id-${i}`,
+  plugin() {
+    if (this.type === 'plugin') {
+      return this.hostCatalog.plugin;
+    }
+  },
 });
diff --git a/addons/api/mirage/generated/factories/host.js b/addons/api/mirage/generated/factories/host.js
index 58e679e7b..7c6e5a24d 100644
--- a/addons/api/mirage/generated/factories/host.js
+++ b/addons/api/mirage/generated/factories/host.js
@@ -13,9 +13,4 @@ export default Factory.extend({
   attributes: () => {
     return { address: internet.ipv6() };
   },
-  plugin: function () {
-    if (this.type === 'plugin') {
-      return this.hostCatalog.plugin;
-    }
-  },
 });