Update test to catch off-by-one