fix sometimes failing tests issue