diff --git a/frontend/src/Visualization.tsx b/frontend/src/Visualization.tsx
index cd90a98..3b9b200 100644
--- a/frontend/src/Visualization.tsx
+++ b/frontend/src/Visualization.tsx
@@ -42,6 +42,7 @@ const Visualization: React.FC = () => {
         setIsHalt(false);
         setMemory(new Array(0).fill({ status: CellStatus.Free }));
         setEventLogs([[SUGGEST_INIT_LOG_ENTRY, undefined]]);
+        clearHighlightedCells();
     }
 
     const handleRestart = () => {