modify script

Change-Id: I2d063c1970b1cf6d65a618355930c82dbce1b9bc