Change the class name for drush