Corrected broken test