Fix the broken test