Fix build in python 3