diff --git a/src/main/java/ai/nets/samj/models/PythonMethods.java b/src/main/java/ai/nets/samj/models/PythonMethods.java
index fc42f52..e5b31d7 100644
--- a/src/main/java/ai/nets/samj/models/PythonMethods.java
+++ b/src/main/java/ai/nets/samj/models/PythonMethods.java
@@ -109,7 +109,7 @@ public class PythonMethods {
 	 * Run-Length Encoding (RLE) algorithm
 	 */
 	protected static String RLE_METHOD = ""
-			+ "def encode_rle(mask: np.ndarray) -> List[int]:" +  System.lineSeparator()
+			+ "def encode_rle(mask):" +  System.lineSeparator()
 			+ "    \"\"\"" +  System.lineSeparator()
 			+ "    Encode a binary mask using Run-Length Encoding (RLE)." +  System.lineSeparator()
 			+ "    " +  System.lineSeparator()