diff --git a/environment.yml b/environment.yml
index 11c7de5..97e7188 100644
--- a/environment.yml
+++ b/environment.yml
@@ -4,7 +4,9 @@ channels:
   - defaults
 dependencies:
   - python=3.10
-  - dask
+  # TODO: Undo version restriction once https://github.com/dmlc/xgboost/issues/11461
+  # has been resolved and released
+  - dask<2025.4.0
   - dask-ml
   - coiled
   - pyarrow
diff --git a/xgboost.ipynb b/xgboost.ipynb
index e710529..38d9580 100644
--- a/xgboost.ipynb
+++ b/xgboost.ipynb
@@ -136,7 +136,7 @@
     "from datetime import datetime\n",
     "\n",
     "import dask.array as da\n",
-    "import xgboost\n",
+    "import xgboost.dask\n",
     "from dask_ml.metrics import mean_squared_error\n",
     "\n",
     "start = datetime.now()\n",
@@ -220,9 +220,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:coiled]",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "conda-env-coiled-py"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -234,7 +234,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.10.17"
   }
  },
  "nbformat": 4,