Changed tests to use logger