One more fix for Python 3 compatibility