Add back debug code