Fix build in python 2