Fix for print in tests