|  | @@ -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
 |