update tests to work properly with directives.