Fix logging
Fix logging