Update build.py a little bit