modify script

Change-Id: Id6cb50d2fe14f8d0da4d8fe7fe500569d912a230