fix error for 1-by-1 case