diff --git a/nbs/simple_icm.py b/nbs/simple_icm.py
index 2c2c182..7ab67d6 100644
--- a/nbs/simple_icm.py
+++ b/nbs/simple_icm.py
@@ -52,28 +52,28 @@ class Config:
     num_seed: int = 8
     max_iters: int = 950  # Small for demo; increase for more
     n_shots: int = 16  # Number of in-context examples
-    model: str = "meta-llama/llama-3.1-8b-instruct"  # Logprobs supported
+    model_id: str = "meta-llama/llama-3.1-8b-instruct"  # Logprobs supported
     provider_whitelist: Tuple[str] = None  # None to let OpenRouter choose
     out_dir: Path = Path("../outputs/icm")  # Directory to save outputs
     log_interval: int = 50  # Log progress every N iterations
 
 C = Config(
-    model="qwen/qwen3-235b-a22b-2507",
+    model_id="qwen/qwen3-235b-a22b-2507",
     provider_whitelist=[ 'Chutes','Nebius',], 
 )
 C.out_dir.mkdir(parents=True, exist_ok=True)
 
 # C = Config(
-#     model="qwen/qwen3-30b-a3b-instruct-2507",
+#     model_id="qwen/qwen3-30b-a3b-instruct-2507",
 #     provider_whitelist=[ 'Chutes','Nebius',],
 # )
 
 # C = Config(
-#     model="meta-llama/llama-3.1-70b-instruct",
+#     model_id="meta-llama/llama-3.1-70b-instruct",
 #     provider_whitelist=[ 'Cerebras','Nebius',],
 # )
 # C = Config(
-#     model="meta-llama/llama-3.1-8b-instruct",
+#     model_id="meta-llama/llama-3.1-8b-instruct",
 #     provider_whitelist=('Cerebras','Nebius',),
 # )
 
diff --git a/pyproject.toml b/pyproject.toml
index 72ce687..81cba5e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,6 @@ authors = [
 requires-python = ">=3.10"
 dependencies = [
     "adjusttext>=1.3.0",
-    "aiocache>=0.12.3",
     "alembic>=1.16.5",
     "altair>=5.5.0",
     "anthropic>=0.69.0",