minor change
fix error for 1-by-1 case