Updated logger to the correct class