fixing the broken test