Use mock_log

Thanks, @manishtomar!