Fix drush compatibility