setup.py: add back missing 'import os'