Fix issue with log output