modify script for assert device set

Change-Id: Ia4498feccae579d69b91d285dbf93838fdac45a1