modify script

Change-Id: I3d0d69e156b0d28c43e75059c83989f5c5d79d05