fix the test too