Remove print statements in tests