Fixed up broken test