Fix deprecation juju ...
Sync tests with implementation