Made correction to setup.py file in pytest