Fix build on Python2