diff --git a/grunt/config/run.js b/grunt/config/run.js
index 696814d0d..bd549b132 100644
--- a/grunt/config/run.js
+++ b/grunt/config/run.js
@@ -45,6 +45,10 @@ var config = {
 
 utils.branches.forEach(function (branch) {
 
+  config['generate_' + branch] = {
+    exec: 'node ./scripts/generate/index.js --branch=' + branch
+  };
+
   config['install_es_' + branch] = {
     exec: './scripts/es.sh install ' + branch,
   };
diff --git a/grunt/tasks.js b/grunt/tasks.js
index 33066552b..c231dd461 100644
--- a/grunt/tasks.js
+++ b/grunt/tasks.js
@@ -9,7 +9,7 @@ module.exports = function (grunt) {
   grunt.registerTask('test', function (branch) {
     var tasks = [
       'jshint',
-      'run:generate',
+      branch ? 'run:generate_' + branch : 'run:generate',
       'mochacov:unit'
     ];