Fix test race