diff --git a/src/backend/base/langflow/components/helpers/id_generator.py b/src/backend/base/langflow/components/helpers/id_generator.py
index ebfbcb2e6..1cfdb16a6 100644
--- a/src/backend/base/langflow/components/helpers/id_generator.py
+++ b/src/backend/base/langflow/components/helpers/id_generator.py
@@ -14,6 +14,7 @@ class IDGeneratorComponent(Component):
     description = "Generates a unique ID."
     icon = "fingerprint"
     name = "IDGenerator"
+    legacy = True
 
     inputs = [
         MessageTextInput(
diff --git a/src/frontend/tests/extended/regression/general-bugs-component-as-tool-shortcut.spec.ts b/src/frontend/tests/extended/regression/general-bugs-component-as-tool-shortcut.spec.ts
index 1ec1e5ccf..dbe3183e4 100644
--- a/src/frontend/tests/extended/regression/general-bugs-component-as-tool-shortcut.spec.ts
+++ b/src/frontend/tests/extended/regression/general-bugs-component-as-tool-shortcut.spec.ts
@@ -14,22 +14,22 @@ test(
       timeout: 100000,
     });
     await page.getByTestId("sidebar-search-input").click();
-    await page.getByTestId("sidebar-search-input").fill("id generator");
+    await page.getByTestId("sidebar-search-input").fill("prompt");
 
     await page
-      .getByTestId("helpersID Generator")
+      .getByTestId("promptsPrompt")
       .hover()
       .then(async () => {
-        await page.getByTestId("add-component-button-id-generator").click();
+        await page.getByTestId("add-component-button-prompt").click();
       });
 
-    await page.waitForSelector('[data-testid="title-ID Generator"]', {
+    await page.waitForSelector('[data-testid="title-Prompt"]', {
       timeout: 3000,
     });
 
     expect(await page.getByText("Toolset", { exact: true }).count()).toBe(0);
 
-    await page.getByTestId("title-ID Generator").click();
+    await page.getByTestId("title-Prompt").click();
     await page.keyboard.press("ControlOrMeta+Shift+m");
 
     await page.waitForSelector('text="Toolset"', {
@@ -39,7 +39,7 @@ test(
       await page.getByText("Toolset", { exact: true }).count(),
     ).toBeGreaterThan(0);
 
-    await page.getByTestId("title-ID Generator").click();
+    await page.getByTestId("title-Prompt").click();
 
     await page.waitForSelector('[data-testid="code-button-modal"]', {
       timeout: 3000,
@@ -61,11 +61,11 @@ test(
     // check if the response is 200
     expect(customComponentResponse?.status()).toBe(200);
 
-    await page.waitForSelector('[data-testid="title-ID Generator"]', {
+    await page.waitForSelector('[data-testid="title-Prompt"]', {
       timeout: 3000,
     });
 
-    await page.getByTestId("title-ID Generator").click();
+    await page.getByTestId("title-Prompt").click();
     await page.keyboard.press("ControlOrMeta+Shift+m");
 
     expect(await page.getByText("Toolset", { exact: true }).count()).toBe(0);