diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index a4b385c..0a5e09b 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.3.1
+current_version = 0.4.0
 commit = True
 tag = True
 
@@ -8,4 +8,3 @@ tag = True
 [bumpversion:file:detectree/__init__.py]
 search = __version__ = "{current_version}"
 replace = __version__ = "{new_version}"
-
diff --git a/detectree/__init__.py b/detectree/__init__.py
index 9097568..27c21e6 100644
--- a/detectree/__init__.py
+++ b/detectree/__init__.py
@@ -4,4 +4,4 @@
 from .train_test_split import TrainingSelector
 from .utils import split_into_tiles
 
-__version__ = "0.3.1"
+__version__ = "0.4.0"
diff --git a/setup.py b/setup.py
index fe57309..73554f2 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
 
 from setuptools import find_packages, setup
 
-__version__ = "0.3.1"
+__version__ = "0.4.0"
 
 classifiers = [
     "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",