diff --git a/apps/enp/views.py b/apps/enp/views.py index f862c60f..225f8ee1 100644 --- a/apps/enp/views.py +++ b/apps/enp/views.py @@ -30,9 +30,9 @@ class DrainViewSet(CustomModelViewSet): equip_data = self.request.query_params.get("equip_data", "no") if self.request.method == "GET": if equip_data == "base": - return DrainWithEquipEnpSerializer - elif equip_data == "enp": return DrainWithEquipBaseSerializer + elif equip_data == "enp": + return DrainWithEquipEnpSerializer return super().get_serializer_class() @swagger_auto_schema(