Tweak to directives - element is already a jQuery object