Fix missing import of os