Fix logging calls