Updates a couple broken tests