Just some clean up on the directive.