updating directives that has a small bug