small fixes, if you can pls merge and release to PyPI