Removing debug puts statement from tests