Fixed py.test run.