Fixed py3 error