Fixed error with user input