diff --git a/directive/index.js b/directive/index.js
index 257a4b19a..5ac3c7c0e 100644
--- a/directive/index.js
+++ b/directive/index.js
@@ -1,7 +1,7 @@
 'use strict';
 var yeoman = require('yeoman-generator');
 
-var Generator = yeoman.generators.Base.extend({
+var Generator = yeoman.Base.extend({
   compose: function() {
     this.composeWith('ng-component:directive', {arguments: this.arguments}, { local: require.resolve('generator-ng-component/directive') });
   }