Fix error in build.py