change logic to find toggle buttong due to UI change Test: locally bug: 205204533 Change-Id: If82eb3db6974134aba40c868404f90928ee5b623