Remove wait off-by-one error