Fix issue for python3