Fixing logger issue