Fix test issue