diff --git a/src/components/CustomStateCellRender/CustomStateCellRender.tsx b/src/components/CustomStateCellRender/CustomStateCellRender.tsx
index b7a095e..6f4ae77 100644
--- a/src/components/CustomStateCellRender/CustomStateCellRender.tsx
+++ b/src/components/CustomStateCellRender/CustomStateCellRender.tsx
@@ -1,4 +1,4 @@
-import { Cell } from "@codex-storage/marketplace-ui-components";
+import { Cell, Tooltip } from "@codex-storage/marketplace-ui-components";
 import PurchaseStateIcon from "../../assets/icons/purchases-state-pending.svg?react";
 import SuccessCircleIcon from "../../assets/icons/success-circle.svg?react";
 import ErrorCircleIcon from "../../assets/icons/error-circle.svg?react";
@@ -8,7 +8,7 @@ type Props = {
   message: string | undefined;
 };
 
-export const CustomStateCellRender = ({ state }: Props) => {
+export const CustomStateCellRender = ({ state, message }: Props) => {
   const icons = {
     pending: PurchaseStateIcon,
     submitted: PurchaseStateIcon,
@@ -32,14 +32,13 @@ export const CustomStateCellRender = ({ state }: Props) => {
   return (
     <Cell>
       <p className={"cell-state" + states[state as keyof typeof states]}>
-        {/* {message ? (
+        {message ? (
           <Tooltip message={message}>
-            <Icon width={32} className="cell-stateIcon" />
+            <Icon width={20} className="cell-stateIcon" />
           </Tooltip>
         ) : (
-          <Icon width={32} className="cell-stateIcon" />
-        )} */}
-        <Icon width={20} />
+          <Icon width={20} className="cell-stateIcon" />
+        )}
       </p>
     </Cell>
   );