Fixed broken test, added log mock