diff --git a/src/components/StateLabel.jsx b/src/components/StateLabel.jsx
index 7fabc934..46c5924b 100644
--- a/src/components/StateLabel.jsx
+++ b/src/components/StateLabel.jsx
@@ -11,12 +11,10 @@ const StateLabel = ({ commentIndex, className }) => {
   useEffect(() => {
     const timer = setTimeout(() => setIsLoading(false), 2000);
     return () => clearTimeout(timer);
-  }, []);
-
-  if (commentIndex === undefined) return null;
+  }, [commentIndex]);
 
   return (
-    commentIndex && stateString !== "Succeeded" ? (
+    commentIndex !== undefined && stateString !== "Succeeded" ? (
       (comment.state === "failed") ? (
         null
       ) : (