Allow custom class on helper

ah, I see what you are saying - you are right. Will update now.