Removing debug code from tests