Add import os to stop failing