Fixes tests failing due to logging changes