Fix for change to Wealdtech core