modify script

Change-Id: Id4991cd3e497ae70aed4c0e888f77d05dd58d9e5