build.py: fix import