Drush compatibility fixed