modify script

Change-Id: Ic6a20b5e5120f4cb98cd99d4a4058d78fd416abe