Allow custom class on helper