Fix test_sms.py

Anther attempt to fix this test was done in pull #1044. 
Any other locator I tried in init did not work, nor any other wait.

v1-train -> Merge.
master -> No.