Fix broken tests from logging