fixed the broken test