Fix off-by-one in unit test