diff --git a/apps/dataset/serializers/dataset_serializers.py b/apps/dataset/serializers/dataset_serializers.py
index 7fdd1b4c8..bc00de6f7 100644
--- a/apps/dataset/serializers/dataset_serializers.py
+++ b/apps/dataset/serializers/dataset_serializers.py
@@ -463,6 +463,7 @@ class DataSetSerializers(serializers.ModelSerializer):
             dataset = DataSet(
                 **{'id': dataset_id, 'name': instance.get("name"), 'desc': instance.get('desc'), 'user_id': user_id,
                    'type': Type.web,
+                   'embedding_mode_id': instance.get('embedding_mode_id'),
                    'meta': {'source_url': instance.get('source_url'), 'selector': instance.get('selector'),
                             'embedding_mode_id': instance.get('embedding_mode_id')}})
             dataset.save()