Updated directive to use the renamed provider