Fix test logging