Fix for python3