Adding some output to fix UT