modify script

Change-Id: Ib02c1c1b56d1f20d1f2fba153d2de1ed19098f4d