diff --git a/algorithms/dqi/dqi_max_xorsat.ipynb b/algorithms/dqi/dqi_max_xorsat.ipynb
index 519c5326..de056647 100644
--- a/algorithms/dqi/dqi_max_xorsat.ipynb
+++ b/algorithms/dqi/dqi_max_xorsat.ipynb
@@ -305,21 +305,17 @@
     "        )\n",
     "\n",
     "\n",
-    "@qfunc\n",
+    "@qfunc(generative=True)\n",
     "def prepare_dick_state_unary_input(max_k: CInt, qvar: QArray[QBit]):\n",
     "    \"\"\"\n",
     "    assumes the input is encoded in qvar in unary encoding. should work for every value\n",
     "    smaller than max_k\n",
     "    \"\"\"\n",
-    "    if_(\n",
-    "        qvar.len > 1,\n",
-    "        lambda: [\n",
-    "            _dicke_split_cycle_shift(max_k, qvar),\n",
-    "            prepare_dick_state_unary_input(\n",
-    "                qmin(max_k, qvar.len - 2), qvar[1 : qvar.len]\n",
-    "            ),\n",
-    "        ],\n",
-    "    )\n",
+    "    if qvar.len > 1:\n",
+    "        _dicke_split_cycle_shift(max_k, qvar)\n",
+    "        prepare_dick_state_unary_input(\n",
+    "            qmin(max_k, qvar.len - 2), qvar[1 : qvar.len]\n",
+    "        )\n",
     "\n",
     "\n",
     "@qfunc\n",