fix the broken test