Python3 fix with generator.next, use next()