Fix build script to be same as wheel docs