Removes debug code from tests