Fix logging issue (#8)