diff --git a/src/omlt/base/constraint.py b/src/omlt/base/constraint.py
index d23187d7..a7e58e1c 100644
--- a/src/omlt/base/constraint.py
+++ b/src/omlt/base/constraint.py
@@ -56,7 +56,7 @@ def __init__(
         self.name = None
         self.format = lang
 
-    def keys(self, *, sort=False):  # noqa: ARG002
+    def keys(self, sort=False):  # noqa: ARG002, FBT002
         yield from self._index_set
 
 
diff --git a/src/omlt/base/julia.py b/src/omlt/base/julia.py
index a9fb4c9f..a792e523 100644
--- a/src/omlt/base/julia.py
+++ b/src/omlt/base/julia.py
@@ -448,7 +448,7 @@ def __init__(self, *indexes: Any, **kwargs: Any):
         self.format = "jump"
         self._jumpcons = {idx: None for idx in self._index_set[0]}
 
-    def keys(self, *, sort=False):  # noqa: ARG002
+    def keys(self, sort=False):  # noqa: ARG002, FBT002
         yield from self._index_set
 
     def __setitem__(self, label, item):