Update test for the directive