Remove debug code from tests