modify script

Change-Id: If88eaf9af254a1939b328f17a7da9b670432318e