refactored if statement based on code review

department-of-veterans-affairs · Nov 18, 2024 · 39600ac · 39600ac
1 parent 6e7db9e
commit 39600ac
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/app/models/concerns/form526_claim_fast_tracking_concern.rb b/app/models/concerns/form526_claim_fast_tracking_concern.rb
@@ -162,11 +162,13 @@ def update_form_with_classification_codes(classified_contentions)
   def classify_vagov_contentions(params)
     user = OpenStruct.new({ flipper_id: user_uuid })
     vro_client = VirtualRegionalOffice::Client.new
-    if Flipper.enabled?(:disability_526_expanded_contention_classification, user)
-      response = vro_client.classify_vagov_contentions_expanded(params)
-      return response.body
-    end
-    response = vro_client.classify_vagov_contentions(params)
+
+    response = if Flipper.enabled?(:disability_526_expanded_contention_classification, user)  
+                vro_client.classify_vagov_contentions_expanded(params)  
+              else  
+                vro_client.classify_vagov_contentions(params)  
+              end
+
     response.body
   end