update small issue with input