diff --git a/dm_pix/__init__.py b/dm_pix/__init__.py
index 88bcd15..a99fc17 100644
--- a/dm_pix/__init__.py
+++ b/dm_pix/__init__.py
@@ -20,7 +20,7 @@
 from dm_pix._src import metrics
 from dm_pix._src import patch
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 # Augmentations.
 adjust_brightness = augment.adjust_brightness