Fix the test (temporarly)