fix pwd

add missing `import os`