Re-add os.path import