Fix Logger functionality