diff --git a/pysr/test/test_main.py b/pysr/test/test_main.py
index 52772042..2f5f440a 100644
--- a/pysr/test/test_main.py
+++ b/pysr/test/test_main.py
@@ -876,8 +876,14 @@ def test_scikit_learn_compatibility(self):
         check_generator = check_estimator(model, generate_only=True)
         exception_messages = []
         for _, check in check_generator:
-            if check.func.__name__ == "check_complex_data":
-                # We can use complex data, so avoid this check.
+            if check.func.__name__ in {
+                # We can use complex data, so avoid this check
+                "check_complex_data",
+                # We handle kwargs manually, so skip this check
+                "check_do_not_raise_errors_in_init_or_set_params",
+                # TODO:
+                "check_n_features_in_after_fitting",
+            }:
                 continue
             try:
                 with warnings.catch_warnings():