Fixing small directive issue.