py.test: fix test