Updated logging for tests