diff --git a/README.md b/README.md
index eb25c29..57ac7ee 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 # mbr 🔥
-![Main](https://github.com/ZurichNLP/mbr/workflows/unittest/badge.svg?branch=main)
+![Main](https://github.com/ZurichNLP/mbr/workflows/unittest/badge.svg)
 [![PyPI](https://img.shields.io/pypi/v/mbr)](https://pypi.python.org/pypi/mbr/)
 
 **mbr** adds Sampling-based Minimum Bayes Risk decoding to [Hugging Face transformers](https://github.com/huggingface/transformers). Originally proposed by [Eikema & Aziz (2022)](https://aclanthology.org/2022.emnlp-main.754/), this technique is a risk-minimizing algorithm for generating text with a language model.
diff --git a/pyproject.toml b/pyproject.toml
index 624f330..17f1b31 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "mbr"
-version = "0.0.1"
+version = "0.1.0"
 authors = [
   { name="Jannis Vamvas", email="vamvas@cl.uzh.ch" },
 ]
diff --git a/src/mbr/__init__.py b/src/mbr/__init__.py
index 004efb5..88ba904 100644
--- a/src/mbr/__init__.py
+++ b/src/mbr/__init__.py
@@ -4,4 +4,4 @@
 from mbr.modeling import MBR
 
 
-__version__ = "0.0.1"
+__version__ = "0.1.0"