Fix InvalidArgumentException