Small fix for python3