Allow to pass a callback to the element helper