Fix input bug