diff --git a/dm_pix/__init__.py b/dm_pix/__init__.py
index e10b073..4349f50 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.4.0"
+__version__ = "0.4.1"
 
 # Augmentations.
 adjust_brightness = augment.adjust_brightness