Fixed custom directive issue.