Update tests to new logger API