Python 2 bug fix
Bump version to `0.3.1`