fix test in the setup.py