Fixes small python3 compatibility issue