diff --git a/Gruntfile.js b/Gruntfile.js
index fe69b40d3..ca5178fc7 100644
--- a/Gruntfile.js
+++ b/Gruntfile.js
@@ -3,7 +3,7 @@
 var shell = require('shelljs');
 var child_process = require('child_process');
 var Q = require('q');
-var helpers = require('yeoman-generator').test;
+var helpers = require('yeoman-test');
 var fs = require('fs');
 var path = require('path');
 
diff --git a/package.json b/package.json
index 52ea37585..55c211fc1 100644
--- a/package.json
+++ b/package.json
@@ -40,7 +40,7 @@
     "glob": "^6.0.1",
     "lodash": "^3.10.1",
     "underscore.string": "^3.1.1",
-    "yeoman-generator": "~0.20.3",
+    "yeoman-generator": "^0.22.5",
     "yeoman-welcome": "^1.0.1"
   },
   "devDependencies": {
@@ -59,7 +59,8 @@
     "q": "^1.0.1",
     "recursive-readdir": "^1.2.0",
     "shelljs": "^0.5.3",
-    "yeoman-assert": "^2.0.0"
+    "yeoman-assert": "^2.0.0",
+    "yeoman-test": "^1.1.0"
   },
   "engines": {
     "node": "^4.2.3",
diff --git a/test/test-file-creation.js b/test/test-file-creation.js
index 3ffa6fbda..f4da5a772 100644
--- a/test/test-file-creation.js
+++ b/test/test-file-creation.js
@@ -3,7 +3,7 @@
 var path = require('path');
 var fs = require('fs');
 var exec = require('child_process').exec;
-var helpers = require('yeoman-generator').test;
+var helpers = require('yeoman-test');
 var assert = require('yeoman-assert');
 var chai = require('chai');
 var expect = chai.expect;