Updating also the test.