Fix test code.