Fix ImportError on running build.py