diff --git a/addons/api/tests/unit/models/host-catalog-test.js b/addons/api/tests/unit/models/host-catalog-test.js
index cf1cc94d5..84adfd1b8 100644
--- a/addons/api/tests/unit/models/host-catalog-test.js
+++ b/addons/api/tests/unit/models/host-catalog-test.js
@@ -79,14 +79,11 @@ module('Unit | Model | host catalog', function (hooks) {
     assert.expect(3);
     const store = this.owner.lookup('service:store');
     const modelPlugin = store.createRecord('host-catalog', {
-      type: 'plugin',
-      plugin: { name: 'Test name' },
+      compositeType: 'aws'
     });
     const modelStatic = store.createRecord('host-catalog', {
-      type: 'static',
+      compositeType: 'static',
     });
-    modelPlugin.set('compositeType', 'aws');
-    modelStatic.set('compositeType', 'static');
 
     assert.equal(modelPlugin.type, 'plugin');
     assert.equal(modelPlugin.plugin.name, 'aws');