diff --git a/bitbucket-pipelines.yml b/bitbucket-pipelines.yml
index 10af031..3fae17d 100644
--- a/bitbucket-pipelines.yml
+++ b/bitbucket-pipelines.yml
@@ -3,7 +3,7 @@
 # Only use spaces to indent your .yml configuration.
 # -----
 # You can specify a custom docker image from Docker Hub as your build environment.
-image: nvidia/cuda
+image: nvidia/cuda:11.0-devel-ubuntu20.04
 
 # JP notes:
 # 1) We need a sufficiently recent GPU to run the code (needs to support CUDA 10 + NVIDIA drivers 418)
@@ -26,7 +26,7 @@ pipelines:
           - apt-get update
           - apt-get install -y apt-transport-https ca-certificates gnupg software-properties-common wget
           - wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | gpg --dearmor - | tee /etc/apt/trusted.gpg.d/kitware.gpg >/dev/null
-          - apt-add-repository 'deb https://apt.kitware.com/ubuntu/ bionic main'
+          - apt-add-repository 'deb https://apt.kitware.com/ubuntu/ focal main'
           - apt-get update
           - apt-get install -y cmake flex bison openmpi-bin libopenmpi-dev gfortran
           - cmake -DDSL_MODULE_DIR="acc/mhd_solver" -DBUILD_STANDALONE=ON -DBUILD_UTILS=ON -DBUILD_RT_VISUALIZATION=OFF -DBUILD_SAMPLES=ON -DDOUBLE_PRECISION=OFF -DMULTIGPU_ENABLED=ON -DMPI_ENABLED=OFF .. # Single precision