fixed off-by-1 in inference