diff --git a/packages/components/nodes/cache/MomentoCache/MomentoCache.ts b/packages/components/nodes/cache/MomentoCache/MomentoCache.ts
index 37d909416..722f7340a 100644
--- a/packages/components/nodes/cache/MomentoCache/MomentoCache.ts
+++ b/packages/components/nodes/cache/MomentoCache/MomentoCache.ts
@@ -21,7 +21,7 @@ class MomentoCache implements INode {
         this.type = 'Cache'
         this.icon = 'momento.png'
         this.category = 'Cache'
-        this.baseClasses = [this.type, ...getBaseClasses(MomentoCache)]
+        this.baseClasses = [this.type, ...getBaseClasses(LangchainMomentoCache)]
         this.credential = {
             label: 'Connect Credential',
             name: 'credential',