Update test code to remove duplicated codes