[redesign] Change test code