Fix input args