Merge branch 'master' of github.com:AF83/stif-boiv

1 files changed, 6 insertions, 3 deletions

diff --git a/app/controllers/compliance_controls_controller.rb b/app/controllers/compliance_controls_controller.rb
index be8b773e2..4dd7e42b2 100644
--- a/app/controllers/compliance_controls_controller.rb
+++ b/app/controllers/compliance_controls_controller.rb
@@ -7,9 +7,12 @@ class ComplianceControlsController < BreadcrumbController
   end
 
   def new
-    redirect_to(action: :select_type) unless params[:sti_class]
-    @compliance_control_set = parent
-    @compliance_control     = params[:sti_class].constantize.new
+    if params[:sti_class]
+      @compliance_control_set = parent
+      @compliance_control     = params[:sti_class].constantize.new
+    else
+      redirect_to(action: :select_type)
+    end
   end
 
   private