Fix bug with completion for custom directive.