modify script

Change-Id: I853b08345308ae85477654b8a72fe8b5c9cac3ee