Added missing import in stt.py