Add test for helper class