lib/acquia_toolbelt/cli.rb in acquia_toolbelt-2.0.0 vs lib/acquia_toolbelt/cli.rb in acquia_toolbelt-2.0.1

@@ -25,11 +25,12 @@
       ui = AcquiaToolbelt::CLI::UI.new
 
       # Use a custom symbol to separate the commands. Useful for rake styled
       # commands.
       if given_args[0].include? ":"
-        given_args = given_args.flat_map {|e| e.split(':')}
+        commands = given_args.shift.split(":")
+        given_args = given_args.unshift(commands).flatten
       end
 
       super(given_args, {:shell => ui}.merge(config))
     rescue AcquiaToolbelt::Error
       ui.print_exception(e)
@@ -97,6 +98,6 @@
 
     # Environments.
     desc "environments", ""
     subcommand "environments", AcquiaToolbelt::CLI::Environments
   end
-end
\ No newline at end of file
+end