Add missing register