Before add the custom directive