Added wait to fix test_sms.py

master -> Don't merge.
v1-train -> Merge.