Corrected python 3 error