Fix test stability