diff --git a/dask_sql.Dockerfile b/dask_sql.Dockerfile
index 09ee41f..77293fc 100644
--- a/dask_sql.Dockerfile
+++ b/dask_sql.Dockerfile
@@ -22,6 +22,8 @@ RUN gpuci_mamba_retry env create -n dask_sql --file /dask_sql_environment.yaml
 RUN gpuci_mamba_retry install -y -n dask_sql -c rapidsai -c rapidsai-nightly -c nvidia -c conda-forge \
     cudatoolkit=$CUDA_VER \
     cudf=$RAPIDS_VER \
+    cuml=$RAPIDS_VER \
+    xgboost \
     dask-cudf=$RAPIDS_VER \
     dask-cuda=$RAPIDS_VER \
     numpy=$NUMPY_VER \