Fix the test (it works)