Fix input loop error