Fix py.test dependency