diff --git a/miplearn/solvers/tests/test_learning_solver.py b/miplearn/solvers/tests/test_learning_solver.py
index ca9719e..ceed608 100644
--- a/miplearn/solvers/tests/test_learning_solver.py
+++ b/miplearn/solvers/tests/test_learning_solver.py
@@ -64,4 +64,4 @@ def test_add_components():
     solver.add(LazyConstraintsComponent())
     solver.add(LazyConstraintsComponent())
     assert len(solver.components) == 1
-    assert "BranchPriorityComponent" in solver.components
+    assert "LazyConstraintsComponent" in solver.components