build.py: fix python import