diff --git a/Project.toml b/Project.toml
index 27f18acf..aa053119 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "Ripserer"
 uuid = "aa79e827-bd0b-42a8-9f10-2b302677a641"
 authors = ["mtsch <matijacufar@gmail.com>"]
-version = "0.16.4"
+version = "0.16.5"
 
 [deps]
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
@@ -26,7 +26,7 @@ DataStructures = "0.17, 0.18"
 Distances = "0.8, 0.9, 0.10"
 IterTools = "1"
 LightGraphs = "1.3.3"
-MLJModelInterface = "^0.3.5"
+MLJModelInterface = "^0.3.5, 0.4"
 MiniQhull = "0.2"
 PersistenceDiagrams = "0.9"
 ProgressMeter = "1"