fix input bug