Add missing controller