diff --git a/tests/test_features.py b/tests/test_features.py
index ac213c8..9aab95b 100644
--- a/tests/test_features.py
+++ b/tests/test_features.py
@@ -25,7 +25,7 @@ def test_knapsack() -> None:
     solver.set_instance(instance, model)
     solver.solve_lp()
 
-    features = FeaturesExtractor(solver).extract(instance)
+    features = FeaturesExtractor().extract(instance, solver)
     assert features.variables is not None
     assert features.constraints is not None
     assert features.instance is not None