modify script

Change-Id: I1df498089be3dca54c06af83aa291337a717b816