Script modification

Change-Id: Iee547a65b51664bdbb55c55828c90d2538f4b1b6