minor fix input bug