Remove prints in test function