modify script

Change-Id: I24b6e7f42785b904bad99a8bbcdc3575cc647847