modify script

Change-Id: I14eab8569091782a02ba560724b7d1df422610df