Replace console.log after modifying it for count test