Add support for other distro for setup.py
Fix unit test