diff --git a/docs/overrides/main.html b/docs/overrides/main.html
index 69fa1145..fffdca2e 100644
--- a/docs/overrides/main.html
+++ b/docs/overrides/main.html
@@ -5,7 +5,7 @@
Navigate the site here!
- v0.2.2 is released!
+ v0.2.3 is released!
{% include ".icons/material/head-question.svg" %}
diff --git a/src/hssm/hssm.py b/src/hssm/hssm.py
index 6c51756c..5934cbe6 100644
--- a/src/hssm/hssm.py
+++ b/src/hssm/hssm.py
@@ -1505,7 +1505,9 @@ def _make_model_distribution(self) -> type[pm.Distribution]:
bounds=self.bounds,
lapse=self.lapse,
extra_fields=(
- None if not self.extra_fields else deepcopy(self.extra_fields)
+ None
+ if not self.extra_fields
+ else [deepcopy(self.data[field].values) for field in self.extra_fields]
),
)