diff --git a/lib/mistral_rb.rb b/lib/mistral_rb.rb
index d23d20d..508483e 100644
--- a/lib/mistral_rb.rb
+++ b/lib/mistral_rb.rb
@@ -20,7 +20,7 @@ def initialize(api_key: ENV["MISTRAL_API_KEY"], base_uri: "https://api.mistral.a
     self.class.base_uri base_uri
   end
 
-  def create_chat_completion(model:, messages:, temperature: 0.7, top_p: 1, max_tokens: nil, stream: false, safe_mode: false, random_seed: nil)
+  def create_chat_completion(model:, messages:, temperature: 0.7, top_p: 1, max_tokens: nil, stream: false, safe_prompt: false, random_seed: nil)
     body = {
       model: model,
       messages: messages,
@@ -28,7 +28,7 @@ def create_chat_completion(model:, messages:, temperature: 0.7, top_p: 1, max_to
       top_p: top_p,
       max_tokens: max_tokens,
       stream: stream,
-      safe_mode: safe_mode,
+      safe_prompt: safe_prompt,
       random_seed: random_seed
     }.compact.to_json