Update test code.