Fix in order to work with Python v. 3