Remove print statements from tests