Add a small margin to a flaky test