diff --git a/lib/omniai/chat/choice.rb b/lib/omniai/chat/choice.rb
index 7988afd..eb77173 100644
--- a/lib/omniai/chat/choice.rb
+++ b/lib/omniai/chat/choice.rb
@@ -12,7 +12,6 @@ class Choice
 
       # @param message [Message]
       # @param index [Integer]
-      # @param tool_call_list [Array<ToolCall>]
       def initialize(message:, index: 0)
         @message = message
         @index = index
diff --git a/lib/omniai/chat/tool_call_result.rb b/lib/omniai/chat/tool_call_result.rb
index 0950e1c..7eaf6f8 100644
--- a/lib/omniai/chat/tool_call_result.rb
+++ b/lib/omniai/chat/tool_call_result.rb
@@ -11,7 +11,7 @@ class ToolCallResult
       attr_accessor :tool_call_id
 
       # @param content [Object]
-      # @param tool_call [ToolCall]
+      # @param tool_call_id [String]
       def initialize(content:, tool_call_id:)
         @content = content
         @tool_call_id = tool_call_id