Fix logging
Fixed logging