Fix build for python3