Slight test modification