Python 2.7 compatibility fix