Fix test oops