Fixed logging with the unit tests