fix broken test