templates/layout.erb in rails_wizard-0.1.2 vs templates/layout.erb in rails_wizard-0.1.3

@@ -36,7 +36,10 @@
 # >-----------------------------[ Run Bundler ]-------------------------------<
 
 say_wizard "Running Bundler install. This will take a while."
 run 'bundle install'
 say_wizard "Running after Bundler callbacks."
-@after_blocks.each{|b| @current_recipe = b[0]; b[1].call}
-@after_everything_blocks.each{|b| @current_recipe = b[0]; b[1].call}
+@after_blocks.each{|b| config = @configs[b[0]] || {}; @current_recipe = b[0]; b[1].call}
+
+@current_recipe = nil
+say_wizard "Running after everything callbacks."
+@after_everything_blocks.each{|b| config = @configs[b[0]] || {}; @current_recipe = b[0]; b[1].call}