Fix off by one error in sampling code