modify script

Change-Id: Ia98ba982d09a77c41b8ed6ae2bb35c3f29ff7f7b