Updated for change in environment API.

drush-ops · Aug 9, 2016 · d9a131c · d9a131c
1 parent 8aaf148
commit d9a131c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/commands/module_builder/module_builder.drush.inc b/commands/module_builder/module_builder.drush.inc
@@ -25,7 +25,8 @@ function module_builder_drush_init() {
  */
 function module_builder_drush_init_helper() {
   // Set up the MB factory.
-  \DrupalCodeBuilder\Factory::setEnvironmentClass('Drush', drush_drupal_version());
+  \DrupalCodeBuilder\Factory::setEnvironmentLocalClass('Drush')
+    ->setCoreVersionNumber(drush_drupal_version());
 }
 
 /**