Python 3 compatibility change