diff --git a/containers/v3-dockerfile-test-suite-python310 b/containers/v3-dockerfile-test-suite-python310
index c663c9f5dac3332723b1cadeb2278badcc568123..ca54919a2c0ee1233a075b12b89066e42c1fb1e6 100644
--- a/containers/v3-dockerfile-test-suite-python310
+++ b/containers/v3-dockerfile-test-suite-python310
@@ -32,7 +32,7 @@ RUN conda install -n ${conda_env} -c conda-forge scikit-image celerite george
 
 # Install dependencies and samplers
 RUN pip install corner healpy cython tables
-RUN conda install -n ${conda_env} dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc3 pymc nessai -c conda-forge -c pytorch
+RUN conda install -n ${conda_env} dynesty emcee nestle ptemcee pymultinest ultranest cpnest kombine dnest4 zeus-mcmc pytorch theano-pymc pymc nessai -c conda-forge -c pytorch
 
 # Install Polychord
 RUN apt-get update --allow-releaseinfo-change
diff --git a/containers/write_dockerfiles.py b/containers/write_dockerfiles.py
index 45a52003ad7f959d489f5f5cb28510dd8dc0547d..046a329742c8eafe978b5f8787d25491c72b074a 100644
--- a/containers/write_dockerfiles.py
+++ b/containers/write_dockerfiles.py
@@ -20,7 +20,7 @@ conda_sampler_dict = dict(
     python310=[
         "dynesty", "emcee", "nestle", "ptemcee", "pymultinest", "ultranest",
         "cpnest", "kombine", "dnest4", "zeus-mcmc",
-        "pytorch", "theano-pymc3", "pymc", "nessai",
+        "pytorch", "theano-pymc", "pymc", "nessai",
     ]
 )