diff --git a/kaizen/llms/provider.py b/kaizen/llms/provider.py
index 28745c5f..2780c2c0 100644
--- a/kaizen/llms/provider.py
+++ b/kaizen/llms/provider.py
@@ -21,7 +21,7 @@ def __init__(
         self.temperature = temperature
         CONFIG_DATA = ConfigData()
         self.config = CONFIG_DATA.get_config_data()
-        if CONFIG_DATA.get("language_model", {}).get(
+        if self.config.get("language_model", {}).get(
             "enable_observability_logging", False
         ):
             # set callbacks
diff --git a/pyproject.toml b/pyproject.toml
index 36f7460d..8dd9ebbc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kaizen-cloudcode"
-version = "0.1.7"
+version = "0.1.8"
 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"