Update codes for tests.