Fix import on `setup.py`.
v0.1.3 release.