modify script

Change-Id: I404773d567620be6560617d65f01924410c29a88