Fix input grabbing