Update the tests too.