Fix passing atom to mix task :main option

Looks like this code wasn't kept in sync with the Elixir main repo.