diff --git a/bot.py b/bot.py
index af0acfe8..2290602a 100644
--- a/bot.py
+++ b/bot.py
@@ -37,7 +37,7 @@
 # if Neo4j is local, you can go to http://localhost:7474/ to browse the database
 neo4j_graph = Neo4jGraph(url=url, username=username, password=password)
 embeddings, dimension = load_embedding_model(
-    embedding_model_name, config={ollama_base_url: ollama_base_url}, logger=logger
+    embedding_model_name, config={"ollama_base_url": ollama_base_url}, logger=logger
 )
 create_vector_index(neo4j_graph, dimension)
 
diff --git a/chains.py b/chains.py
index ee62302e..bcfa70c1 100644
--- a/chains.py
+++ b/chains.py
@@ -15,7 +15,9 @@
 
 def load_embedding_model(embedding_model_name: str, logger=BaseLogger(), config={}):
     if embedding_model_name == "ollama":
-        embeddings = OllamaEmbeddings(base_url=config.ollama_base_url, model="llama2")
+        embeddings = OllamaEmbeddings(
+            base_url=config["ollama_base_url"], model="llama2"
+        )
         dimension = 4096
         logger.info("Embedding: Using Ollama")
     elif embedding_model_name == "openai":
diff --git a/loader.py b/loader.py
index 65cad8ed..8cc08023 100644
--- a/loader.py
+++ b/loader.py
@@ -23,7 +23,7 @@
 so_api_base_url = "https://api.stackexchange.com/2.3/search/advanced"
 
 embeddings, dimension = load_embedding_model(
-    embedding_model_name, config={ollama_base_url: ollama_base_url}, logger=logger
+    embedding_model_name, config={"ollama_base_url": ollama_base_url}, logger=logger
 )
 
 # if Neo4j is local, you can go to http://localhost:7474/ to browse the database
diff --git a/pdf_bot.py b/pdf_bot.py
index 81bad5d9..fde7772b 100644
--- a/pdf_bot.py
+++ b/pdf_bot.py
@@ -31,7 +31,7 @@
 
 
 embeddings, dimension = load_embedding_model(
-    embedding_model_name, config={ollama_base_url: ollama_base_url}, logger=logger
+    embedding_model_name, config={"ollama_base_url": ollama_base_url}, logger=logger
 )