diff --git a/index.js b/index.js
index 67040db..695ae5a 100644
--- a/index.js
+++ b/index.js
@@ -6,18 +6,17 @@ const { hasPlugin, addPlugin } = require('ember-cli-babel-plugin-helpers');
 module.exports = {
   name: require('./package').name,
 
-  included() {
+  included(parent) {
     this._super.included.apply(this, arguments);
 
-    this.addBabelPlugin();
+    this.addBabelPlugin(parent);
   },
 
-  addBabelPlugin() {
-    let app = this._findHost();
+  addBabelPlugin(parent) {
     let pluginPath = resolve(__dirname, './lib/transpile-modules.js');
 
-    if (!hasPlugin(app, pluginPath)) {
-      addPlugin(app, pluginPath);
+    if (!hasPlugin(parent, pluginPath)) {
+      addPlugin(parent, pluginPath);
     }
   }
 };