Small fix to avoid warning when run with Py3