Fix for python3?