Fixing changes in cli script update