Refactor to be consistent with mina-sidekiq

Fixes a bug: undefined method `set_default' for main:Object (NoMethodError) when starting a rake task. See also https://github.com/Mic92/mina-sidekiq/issues/2. Pulled this fix from another fork (thanks @andywatts!)