Fix occassional test failure