diff --git a/prompting/tasks/inference.py b/prompting/tasks/inference.py
index fefa5b7a..c5766610 100644
--- a/prompting/tasks/inference.py
+++ b/prompting/tasks/inference.py
@@ -61,5 +61,5 @@ def make_reference(self, dataset_entry: ChatEntry) -> str:
             roles=["user"],
             model=self.llm_model,
             sampling_params=SamplingParams(seed=self.seed),
-        )[0]
+        )
         return self.reference
diff --git a/pyproject.toml b/pyproject.toml
index 21dd1f4e..8993cad7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prompting"
-version = "2.12.0"
+version = "2.12.1"
 description = "Subnetwork 1 runs on Bittensor and is maintained by Macrocosmos. It's an effort to create decentralised AI"
 authors = ["Kalei Brady, Dmytro Bobrenko, Felix Quinque, Steffen Cruz"]
 readme = "README.md"