diff --git a/kaizen/llms/provider.py b/kaizen/llms/provider.py
index da81bc01..1d817ea6 100644
--- a/kaizen/llms/provider.py
+++ b/kaizen/llms/provider.py
@@ -74,7 +74,7 @@ def available_tokens(self, message, percentage=0.8):
         )
 
     def get_token_count(self, message):
-        return litellm.token_counter(self.model, message)
+        return litellm.token_counter(model=self.model, text=message)
 
     def update_usage(self, total_usage, current_usage):
         if total_usage is not None:
diff --git a/pyproject.toml b/pyproject.toml
index 17e09fa1..9e43de91 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kaizen-cloudcode"
-version = "0.1.14"
+version = "0.1.15"
 description = "An intelligent coding companion that accelerates your development workflow by providing efficient assistance, enabling you to craft high-quality code more rapidly."
 authors = ["Saurav Panda <saurav.panda@cloudcode.ai>"]
 license = "Apache2.0"