Add helper tests