Fix tests

Tests were broken since parent artefact was upgraded.