Update test codes