Added missing brackets to fix INFRA-135