Add missing code back.