Update test as well.