Some api changes on provider.
Chanded test to work with new api changes