Fixed typo
Fixed Python2 support