fix of limitation. Still not working in 2.X
limitation fixed. Work with Python2 and Python3