Make work with py3.x