Fix for py3