diff --git a/src/main/java/com/ai/da/model/dto/LibraryLevel2TypeUpdateDTO.java b/src/main/java/com/ai/da/model/dto/LibraryLevel2TypeUpdateDTO.java
index c1d7306e..dfb2dba2 100644
--- a/src/main/java/com/ai/da/model/dto/LibraryLevel2TypeUpdateDTO.java
+++ b/src/main/java/com/ai/da/model/dto/LibraryLevel2TypeUpdateDTO.java
@@ -15,7 +15,7 @@ public class LibraryLevel2TypeUpdateDTO {
 
     @NotEmpty(message = "libraryId cannot be empty!")
     @ApiModelProperty("libraryId")
-    private Long libraryId;
+    private List<Long> libraryId;
 
     @NotBlank(message = "level2Type cannot be empty!")
     @ApiModelProperty("level2Type")
diff --git a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
index a57afaf4..35b24038 100644
--- a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
@@ -609,10 +609,9 @@ public class LibraryServiceImpl extends ServiceImpl<LibraryMapper, Library> impl
     @Override
     public Boolean updateLibraryLevel2Type(LibraryLevel2TypeUpdateDTO libraryLevel2TypeUpdateDTO) {
         Library library = new Library();
-        library.setId(libraryLevel2TypeUpdateDTO.getLibraryId());
         library.setLevel2Type(libraryLevel2TypeUpdateDTO.getLevel2Type());
         QueryWrapper<Library> qw = new QueryWrapper<>();
-        qw.lambda().eq(Library::getId, libraryLevel2TypeUpdateDTO.getLibraryId());
+        qw.lambda().in(Library::getId, libraryLevel2TypeUpdateDTO.getLibraryId());
         return libraryMapper.update(library, qw) == 1;
     }