Fix test =)