Fix bug that meant plugin custom states weren't being loaded

1 files changed, 2 insertions, 2 deletions

diff --git a/app/models/info_request.rb b/app/models/info_request.rb
index f7a8f58a2..dcef9e5b5 100644
--- a/app/models/info_request.rb
+++ b/app/models/info_request.rb
@@ -94,7 +94,6 @@ class InfoRequest < ActiveRecord::Base
         'requires_admin',
         'user_withdrawn'
         ]
-
         if @@custom_states_loaded
             states += InfoRequest.theme_extra_states
         end
@@ -108,7 +107,8 @@ class InfoRequest < ActiveRecord::Base
 
     @@custom_states_loaded = false
     begin
-        if !ENV["RAILS_ENV"] == "test"
+        if ENV["RAILS_ENV"] != "test"
+            require 'customstates'
             include InfoRequestCustomStates
             @@custom_states_loaded = true
         end