Add missing include update code