Fix test minor mistake