more updates to tests