src/components/Documents/ManageDocument.jsx

@@ -128,7 +128,7 @@ const ManageDocument = ({ closeModal, Document_Entity, Entity }) => {
       (t) => String(t.id) === String(documentTypeId)
     );
     if (!type) return;
-
+   setSelectedType(type)
     const newSchema = DocumentPayloadSchema({
       isMandatory: type.isMandatory ?? false,
       regexExpression: type.regexExpression ?? null,

src/hooks/masterHook/useMaster.js

@@ -178,7 +178,7 @@ const {
     isError,
     isLoading
   } = useQuery({
-    queryKey: ["Document Type"],
+    queryKey: ["Document Type",category],
     queryFn: async () => {
       const res = await MasterRespository.getDocumentTypes(category)
       return res.data;