Fix uit tests