Fix custom directive