diff --git a/odds/common/llm/cost_collector.py b/odds/common/cost_collector.py
similarity index 100%
rename from odds/common/llm/cost_collector.py
rename to odds/common/cost_collector.py
diff --git a/odds/common/embedder/openai/openai_embedder.py b/odds/common/embedder/openai/openai_embedder.py
index 698958b..b789ff4 100644
--- a/odds/common/embedder/openai/openai_embedder.py
+++ b/odds/common/embedder/openai/openai_embedder.py
@@ -5,7 +5,7 @@
 from ...datatypes import Embedding
 
 from ..embedder import Embedder
-from ...llm.cost_collector import CostCollector
+from ...cost_collector import CostCollector
 from ...config import config
 from ...retry import Retry
 
diff --git a/odds/common/llm/llm_runner.py b/odds/common/llm/llm_runner.py
index 8f509cb..8418f6a 100644
--- a/odds/common/llm/llm_runner.py
+++ b/odds/common/llm/llm_runner.py
@@ -1,6 +1,6 @@
 from .llm_query import LLMQuery
 from .llm_cache import LLMCache
-from .cost_collector import CostCollector
+from ..cost_collector import CostCollector
 
 class LLMRunner:
 
diff --git a/test_assistant.py b/test_assistant.py
index edc8add..75aeb82 100644
--- a/test_assistant.py
+++ b/test_assistant.py
@@ -7,7 +7,7 @@
 from odds.common.store import store
 from odds.common.embedder import embedder
 from odds.common.catalog_repo import catalog_repo
-from odds.common.llm.cost_collector import CostCollector
+from odds.common.cost_collector import CostCollector
 from odds.common.llm.openai.openai_llm_runner import OpenAILLMRunner
 import sqlite3