Remove prints from unittests