Fix for python2