lib/mina/proteus.rb in mina-proteus-0.5.3 vs lib/mina/proteus.rb in mina-proteus-0.5.4

@@ -77,13 +77,17 @@
     file = "#{_stages_dir}/#{fetch(:current_stage)}/#{fetch(:current_app)}.rb"
     load file
   end
 end
 
-_potential_stage = ARGV.first
+_potential_stage = ARGV[0]
 
 if !_stage_file_exists?(_potential_stage) && _stage_file_exists?(_default_stage)
+  set :current_app, ARGV[0]
   invoke _default_stage
+else
+  set :current_stage, ARGV[0]
+  set :current_app, ARGV[1]
 end
 
 namespace :proteus do
   desc 'Create stage and apps files'
   task :init do
\ No newline at end of file