Fixed accidently broken test