update tests so that they are working again