Update tests a little.