diff --git a/frontends/concrete-python/examples/levenshtein_distance/levenshtein_distance.py b/frontends/concrete-python/examples/levenshtein_distance/levenshtein_distance.py
index 12758bba20..6dc57fd73f 100644
--- a/frontends/concrete-python/examples/levenshtein_distance/levenshtein_distance.py
+++ b/frontends/concrete-python/examples/levenshtein_distance/levenshtein_distance.py
@@ -217,12 +217,12 @@ def _compute_in_fhe(self, list_patterns: list, show_distance: bool = False):
 @fhe.module()
 class LevenshsteinModule:
     @fhe.function({"x": "encrypted", "y": "encrypted"})
-    def equal(x: int, y: int):
+    def equal(x, y):
         """Assert equality between two chars of the alphabet."""
         return x == y
 
     @fhe.function({"x": "clear"})
-    def constant(x: int):
+    def constant(x):
         return fhe.zero() + x
 
     @fhe.function(
@@ -234,7 +234,7 @@ def constant(x: int):
             "case_3": "encrypted",
         }
     )
-    def mix(is_equal: bool, if_equal: int, case_1: int, case_2: int, case_3: int):
+    def mix(is_equal, if_equal, case_1, case_2, case_3):
         """Compute the min of (case_1, case_2, case_3), and then return `if_equal` if `is_equal` is
         True, or the min in the other case."""
         min_12 = numpy.minimum(case_1, case_2)