Fix declaration of the directive - allow code written after directive to be executed.