Fix some logging