Fix test_sms.py by updating locators