Update tests to match current logging behavior