Use yield in test