diff --git a/torax/config/config_args.py b/torax/config/config_args.py
index d50607cc..c2278d68 100644
--- a/torax/config/config_args.py
+++ b/torax/config/config_args.py
@@ -70,7 +70,7 @@ def _check(ft):
       if _check(arg):
         return True
   else:
-    return _check(field_type)
+    return _check(field_type)  # pytype: disable=bad-return-type
 
 
 def _is_bool(
@@ -166,7 +166,7 @@ def _check(ft):
       if _check(arg):
         return True
   else:
-    return _check(field_type)
+    return _check(field_type)  # pytype: disable=bad-return-type
 
 
 def _load_from_primitives(