Fix run script for CLI version.