Fix small python3 error