modify script

Change-Id: I94adc9d87830678a66a92b486cb06a4589907ffa