Fix broken test file