Fix input bugs