Add logger to tests as well