diff --git a/src/AI_Command.php b/src/AI_Command.php
index 9394e2c..59e5d44 100644
--- a/src/AI_Command.php
+++ b/src/AI_Command.php
@@ -343,7 +343,7 @@ private function generate_text( $builder, $assoc_args ) {
 				// @phpstan-ignore class.notFound
 				$token_usage[ TokenUsage::KEY_COMPLETION_TOKENS ],
 				// @phpstan-ignore class.notFound
-				$token_usage[ TokenUsage::KEY_TOTAL_TOKENS ],
+				$token_usage[ TokenUsage::KEY_TOTAL_TOKENS ]
 			),
 			'ai'
 		);