diff --git a/deep_river/__version__.py b/deep_river/__version__.py
index c7e7843..de49929 100644
--- a/deep_river/__version__.py
+++ b/deep_river/__version__.py
@@ -1,3 +1,3 @@
-VERSION = (0, 2, 3)
+VERSION = (0, 2, 4)
 
 __version__ = ".".join(map(str, VERSION))  # noqa: F401
diff --git a/setup.py b/setup.py
index 33bb665..21dd8df 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
     "torch~=1.13.0",
     "pandas~=1.3.2",
     "numpy~=1.24.0",
-    "river~=0.16.0",
+    "river~=0.17.0",
     "tqdm~=4.61.2",
     "ordered-set~=4.1.0",
     "torchviz~=0.0.2",