diff --git a/bean/notebooks/sample_quality_report.ipynb b/bean/notebooks/sample_quality_report.ipynb
index 23f49ec..2f96598 100755
--- a/bean/notebooks/sample_quality_report.ipynb
+++ b/bean/notebooks/sample_quality_report.ipynb
@@ -94,7 +94,8 @@
    "source": [
     "if not isinstance(replicate_label, str):\n",
     "    bdata.uns[\"sample_covariates\"] = replicate_label[1:]\n",
-    "bdata.samples[replicate_label] = bdata.samples[replicate_label].astype(str)"
+    "bdata.samples[\"replicate\"] = bdata.samples[replicate_label] = bdata.samples[replicate_label].astype(str)\n",
+    "bdata.samples[\"condition\"] = bdata.samples[condition_label]"
    ]
   },
   {