Fix logging for test