Fix for Python 3.x version as well