Changed the directive restriction from element to attribute