diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 84a6aafd30795df58bfe017ece4d14403dbaecf6..57415e7d6beede82dd272225360a995ce3d9459f 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -123,21 +123,20 @@ python-3.8:
   <<: *unit-test
   needs: ["basic-3.8", "precommits-py3.8"]
   image: containers.ligo.org/lscsoft/bilby/v2-bilby-python38
+
+python-3.9:
+  <<: *unit-test
+  needs: ["basic-3.9", "precommits-py3.9"]
+  image: containers.ligo.org/lscsoft/bilby/v2-bilby-python39
   after_script:
     - coverage html
     - coverage xml
-    - coverage-badge -o coverage_badge.svg -f
-
   artifacts:
     reports:
       cobertura: coverage.xml
     paths:
       - htmlcov/
-
-python-3.9:
-  <<: *unit-test
-  needs: ["basic-3.9", "precommits-py3.9"]
-  image: containers.ligo.org/lscsoft/bilby/v2-bilby-python39
+    expire_in: 30 days
 
 # add back when 3.10 image is available
 #python-3.10:
@@ -211,7 +210,6 @@ plotting-python-3.9:
 
 docs:
   stage: docs
-  needs: ["python-3.9"]
   image: containers.ligo.org/lscsoft/bilby/v2-bilby-python39
   script:
     # Make the documentation
@@ -235,11 +233,10 @@ docs:
 
 deploy-docs:
   stage: deploy
-  needs: ["docs"]
+  needs: ["docs", "python-3.9"]
   script:
     - mkdir public/
     - mv htmlcov/ public/
-    - mv coverage_badge.svg public/
     - mv docs/_build/html/* public/
   artifacts:
     paths:
diff --git a/README.rst b/README.rst
index bddce5f89f78ff19960a31d6ca9c07b89943b6b9..37b98613e0fb68572f378cfb9340b53856b3ac93 100644
--- a/README.rst
+++ b/README.rst
@@ -73,7 +73,7 @@ as requested in their associated documentation.
 
 .. |pipeline status| image:: https://git.ligo.org/lscsoft/bilby/badges/master/pipeline.svg
    :target: https://git.ligo.org/lscsoft/bilby/commits/master
-.. |coverage report| image:: https://lscsoft.docs.ligo.org/bilby/coverage_badge.svg
+.. |coverage report| image:: https://git.ligo.org/lscsoft/bilby/badges/master/coverage.svg
    :target: https://lscsoft.docs.ligo.org/bilby/htmlcov/
 .. |pypi| image:: https://badge.fury.io/py/bilby.svg
    :target: https://pypi.org/project/bilby/
diff --git a/test/bilby_mcmc/__init__.py b/test/bilby_mcmc/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391