lib/saasable/middleware.rb in saasable-0.3.2 vs lib/saasable/middleware.rb in saasable-0.3.3

@@ -8,10 +8,12 @@
 
   def call env
     env[:saasable] = {:current_saas => saas_for_host(env["SERVER_NAME"])}
     env[:saasable][:current_saas].activate! if env[:saasable][:current_saas]
 
-    @app.call env
+    @app.call(env).tap do
+      env[:saasable][:current_saas].deactivate! if env[:saasable][:current_saas]
+    end
   end
 
   private
     def saas_for_host hostname
       Saasable::Mongoid::SaasDocument.saas_document.find_by_host!(hostname) rescue nil
\ No newline at end of file