Fix py.test requirement