Trying to fix randomly failing test