diff --git a/dm_pix/__init__.py b/dm_pix/__init__.py
index 0fa5001..f6aed1b 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.3"
+__version__ = "0.3.4"
 
 # Augmentations.
 adjust_brightness = augment.adjust_brightness