Changed solution to match directive name