Remove debug puts in test