Fix previous commit on python 2