Update a few tests