diff --git a/lib/modules/sentiment/post_classifier.rb b/lib/modules/sentiment/post_classifier.rb
index 0b2638f0..a773da91 100644
--- a/lib/modules/sentiment/post_classifier.rb
+++ b/lib/modules/sentiment/post_classifier.rb
@@ -25,7 +25,7 @@ module ::DiscourseAI
                 "#{SiteSetting.ai_sentiment_inference_service_api_endpoint}/api/v1/classify",
                 model,
                 content,
-                SiteSetting.ai_sentiment_inference_service_api_key
+                SiteSetting.ai_sentiment_inference_service_api_key,
               )
 
             store_classification(model, classification)
diff --git a/lib/modules/toxicity/classifier.rb b/lib/modules/toxicity/classifier.rb
index ee7a951e..ba8ca7e0 100644
--- a/lib/modules/toxicity/classifier.rb
+++ b/lib/modules/toxicity/classifier.rb
@@ -26,7 +26,7 @@ module ::DiscourseAI
             "#{SiteSetting.ai_toxicity_inference_service_api_endpoint}/api/v1/classify",
             SiteSetting.ai_toxicity_inference_service_api_model,
             content,
-            SiteSetting.ai_toxicity_inference_service_api_key
+            SiteSetting.ai_toxicity_inference_service_api_key,
           )
         store_classification
         consider_flagging
diff --git a/lib/shared/inference_manager.rb b/lib/shared/inference_manager.rb
index 03f060f8..554cd941 100644
--- a/lib/shared/inference_manager.rb
+++ b/lib/shared/inference_manager.rb
@@ -3,22 +3,11 @@
 module ::DiscourseAI
   class InferenceManager
     def self.perform!(endpoint, model, content, api_key)
+      headers = { "Referer" => Discourse.base_url, "Content-Type" => "application/json" }
 
-      headers = {
-        "Referer" => Discourse.base_url,
-        "Content-Type" => "application/json",
-      }
+      headers["X-API-KEY"] = api_key if api_key.present?
 
-      if api_key.present?
-        headers["X-API-KEY"] = api_key
-      end
-
-      response =
-        Faraday.post(
-          endpoint,
-          { model: model, content: content }.to_json,
-          headers,
-        )
+      response = Faraday.post(endpoint, { model: model, content: content }.to_json, headers)
 
       raise Net::HTTPBadResponse unless response.status == 200