Added missing code for logger