Allow custom class on helper

Same as previous comment.