diff --git a/containers/dockerfile-template b/containers/dockerfile-template
index f4eca79d814edd4ecb9a6ba2e2fbd31bbeafa2ad..af8a61f58a394c7a599609bdf09d09f29b43d3bd 100644
--- a/containers/dockerfile-template
+++ b/containers/dockerfile-template
@@ -37,6 +37,7 @@ RUN conda install -n ${{conda_env}} -c conda-forge ptmcmcsampler
 RUN conda install -n ${{conda_env}} -c conda-forge pytorch
 RUN conda install -n ${{conda_env}} -c conda-forge theano-pymc
 RUN conda install -n ${{conda_env}} -c conda-forge pymc3
+RUN conda install -n ${{conda_env}} -c conda-forge pymc pymc-base
 RUN pip install nessai
 
 # Install Polychord
diff --git a/containers/v3-dockerfile-test-suite-python38 b/containers/v3-dockerfile-test-suite-python38
index 4ee250e8b5792e2c96c78342a974c6a4fd8b97d9..447302e50947cd20b189dbaa7aca578d3201d493 100644
--- a/containers/v3-dockerfile-test-suite-python38
+++ b/containers/v3-dockerfile-test-suite-python38
@@ -39,6 +39,7 @@ RUN conda install -n ${conda_env} -c conda-forge ptmcmcsampler
 RUN conda install -n ${conda_env} -c conda-forge pytorch
 RUN conda install -n ${conda_env} -c conda-forge theano-pymc
 RUN conda install -n ${conda_env} -c conda-forge pymc3
+RUN conda install -n ${conda_env} -c conda-forge pymc pymc-base
 RUN pip install nessai
 
 # Install Polychord
diff --git a/containers/v3-dockerfile-test-suite-python39 b/containers/v3-dockerfile-test-suite-python39
index 218b3d3e24b3b65b278a95958ffb0444b4f85d7c..2d5e22444f254e9cef2c31eb1ae4b78ba22c68d1 100644
--- a/containers/v3-dockerfile-test-suite-python39
+++ b/containers/v3-dockerfile-test-suite-python39
@@ -39,6 +39,7 @@ RUN conda install -n ${conda_env} -c conda-forge ptmcmcsampler
 RUN conda install -n ${conda_env} -c conda-forge pytorch
 RUN conda install -n ${conda_env} -c conda-forge theano-pymc
 RUN conda install -n ${conda_env} -c conda-forge pymc3
+RUN conda install -n ${conda_env} -c conda-forge pymc pymc-base
 RUN pip install nessai
 
 # Install Polychord