Fix test partially