diff --git a/frontend/src/components/Final/Final.test.js b/frontend/src/components/Final/Final.test.js
index fc4c8213b..e2c6f4a58 100644
--- a/frontend/src/components/Final/Final.test.js
+++ b/frontend/src/components/Final/Final.test.js
@@ -1,4 +1,5 @@
 import React from 'react';
+import { vi } from 'vitest';
 import { render, screen, fireEvent, waitFor } from '@testing-library/react';
 import { BrowserRouter, Router } from 'react-router-dom';
 import { createMemoryHistory } from 'history'
diff --git a/frontend/src/components/PlayButton/PlayCard.test.js b/frontend/src/components/PlayButton/PlayCard.test.js
index 8c9046626..732677539 100644
--- a/frontend/src/components/PlayButton/PlayCard.test.js
+++ b/frontend/src/components/PlayButton/PlayCard.test.js
@@ -1,6 +1,7 @@
 import React from "react";
+import { vi } from "vitest";
 import { render, fireEvent, screen } from "@testing-library/react";
-import PlayCard from "./PlayCard"; // Adjust the path as necessary
+import PlayCard from "./PlayCard";
 
 describe("PlayCard Component Tests", () => {
     const mockOnClick = vi.fn();
diff --git a/frontend/src/components/Playback/Playback.test.js b/frontend/src/components/Playback/Playback.test.js
index e3db9b51b..170544db2 100644
--- a/frontend/src/components/Playback/Playback.test.js
+++ b/frontend/src/components/Playback/Playback.test.js
@@ -1,5 +1,6 @@
 import React from 'react';
-import { render, wait } from '@testing-library/react';
+import { vi } from 'vitest';
+import { render } from '@testing-library/react';
 
 import Playback from './Playback';
 
diff --git a/frontend/src/components/UserFeedback/UserFeedback.test.js b/frontend/src/components/UserFeedback/UserFeedback.test.js
index 8350dbc61..826b12294 100644
--- a/frontend/src/components/UserFeedback/UserFeedback.test.js
+++ b/frontend/src/components/UserFeedback/UserFeedback.test.js
@@ -1,4 +1,5 @@
 import React from 'react';
+import { vi } from 'vitest';
 import { render, fireEvent, waitFor } from '@testing-library/react';
 import UserFeedback from './UserFeedback';
 import { postFeedback } from '../../API';
diff --git a/frontend/src/components/VisualMatchingPairs/VisualMatchingPairs.test.js b/frontend/src/components/VisualMatchingPairs/VisualMatchingPairs.test.js
index 34b4d500f..203560cf1 100644
--- a/frontend/src/components/VisualMatchingPairs/VisualMatchingPairs.test.js
+++ b/frontend/src/components/VisualMatchingPairs/VisualMatchingPairs.test.js
@@ -1,4 +1,5 @@
 import React from 'react';
+import { vi } from 'vitest';
 import { render, fireEvent, waitFor } from '@testing-library/react';
 import VisualMatchingPairs from './VisualMatchingPairs'; // Adjust the import path as necessary