diff --git a/tests/fuzzy/test_functions.py b/tests/fuzzy/test_functions.py
index 09d1f8484c28e8a47d82d747921dcdb764b69b7a..56fe671c8833df067d1f57f70f1650f497d5da15 100644
--- a/tests/fuzzy/test_functions.py
+++ b/tests/fuzzy/test_functions.py
@@ -14,11 +14,6 @@ from tests.fuzzy.init import MAX_EXAMPLES, functionKwargs
 def callWontBreak(drawer, func_name: str):
     func = FUNC_MAP[func_name]
     kwargs = drawer.draw(functionKwargs(func))
-
-    # TODO: workaround until `flag` is explicitly exposed in signature
-    flag = drawer.draw(from_type(float))
-    kwargs.setdefault('flag', flag)
-
     func(**kwargs)