Fix logging import