lib/rails-env.rb in rails-env-0.1.0 vs lib/rails-env.rb in rails-env-0.2.0

@@ -6,10 +6,10 @@
       Rails.env.extend(Extension)
     end
   end
 
   module Extension
-    def on(env, &block)
-      block.call(Rails.configuration) if Rails.env.to_s == env.to_s
+    def on(*envs, &block)
+      block.call(Rails.configuration) if envs.include?(Rails.env.to_sym)
     end
   end
 end