Add --mode option for run command