Fix flaky test

Change test so it isn't checking right on the boundary of an edge case