diff --git a/src/main/java/io/bioimage/modelrunner/model/Model.java b/src/main/java/io/bioimage/modelrunner/model/Model.java
index 8d5abfae..0e1e03b4 100644
--- a/src/main/java/io/bioimage/modelrunner/model/Model.java
+++ b/src/main/java/io/bioimage/modelrunner/model/Model.java
@@ -778,8 +778,8 @@ public boolean isLoaded() {
 	}
 	
 	/**
-	 * Create consumer used to be used with {@link Model} for the methods {@link #runBioimageioModelOnImgLib2WithTiling(List, Consumer)}
-	 * or {@link #runBioimageioModelOnImgLib2WithTiling(List, Map, Consumer)}.
+	 * Create consumer used to be used with {@link Model} for the methods {@link #runBioimageioModelOnImgLib2WithTiling(List, TilingConsumer)}
+	 * or {@link #runBioimageioModelOnImgLib2WithTiling(List, Map, TilingConsumer)}.
 	 * The coonsumer helps to track the number if tiles that have already been processed.
 	 * @return a consumer to track the tiling process
 	 */