modify script

Change-Id: I387cff61500d164312a815f1e6ae41e7ace714b2