fix input size bug