fixing another input code bug