Fixing the build on python 3