Fixed a Python2-ism