BUG: Fixed py3 bug