Add missing variable from new prima code