Fix logging setup