| @@ -200,8 +200,8 @@ class DatasetSegmentApi(DatasetApiResource): | |||
| parser.add_argument('segments', type=dict, required=False, nullable=True, location='json') | |||
| args = parser.parse_args() | |||
| SegmentService.segment_create_args_validate(args['segments'], document) | |||
| segment = SegmentService.update_segment(args['segments'], segment, document, dataset) | |||
| SegmentService.segment_create_args_validate(args, document) | |||
| segment = SegmentService.update_segment(args, segment, document, dataset) | |||
| return { | |||
| 'data': marshal(segment, segment_fields), | |||
| 'doc_form': document.doc_form | |||