Making tests working with setup.py