Fix for python3 compat