Updated to use logger of test class