diff --git a/datahub-web-react/src/app/entityV2/domain/DataProductsTab/DataProductResult.tsx b/datahub-web-react/src/app/entityV2/domain/DataProductsTab/DataProductResult.tsx index 4036be8cca2e73..a959a97c9dd90f 100644 --- a/datahub-web-react/src/app/entityV2/domain/DataProductsTab/DataProductResult.tsx +++ b/datahub-web-react/src/app/entityV2/domain/DataProductsTab/DataProductResult.tsx @@ -7,7 +7,6 @@ import { useEntityRegistryV2 } from '../../../useEntityRegistry'; import { PreviewType } from '../../Entity'; import EditDataProductModal from './EditDataProductModal'; import { REDESIGN_COLORS } from '../../shared/constants'; -import useDeleteEntity from '../../shared/EntityDropdown/useDeleteEntity'; const TransparentButton = styled(Button)` color: ${REDESIGN_COLORS.RED_ERROR}; @@ -62,10 +61,8 @@ export default function DataProductResult({ dataProduct, onUpdateDataProduct, se setDeletedDataProductUrns((currentUrns) => [...currentUrns, dataProduct.urn]); } - const { onDeleteEntity } = useDeleteEntity(dataProduct.urn, dataProduct.type, dataProduct, deleteDataProduct); - function onDeleteDataProduct() { - onDeleteEntity(); + deleteDataProduct(); setTimeout(() => refetch(), 3000); }