fix: added proper check to test for map type

Signed-off-by: Muskan Bararia <[email protected]>
accordproject · Nov 14, 2023 · b030c98 · b030c98
1 parent d86d203
commit b030c98
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/concerto-core/lib/decoratorextractor.js b/packages/concerto-core/lib/decoratorextractor.js
@@ -16,6 +16,7 @@
 
 const ModelManager = require('./modelmanager');
 const ModelUtil = require('./modelutil');
+const { MetaModelNamespace } = require('@accordproject/concerto-metamodel');
 
 /**
  * Utility functions to work with
@@ -331,7 +332,7 @@ class DecoratorExtractor {
             if (this.removeDecoratorsFromModel){
                 decl.decorators = undefined;
             }
-            if (decl.$class.endsWith('.MapDeclaration')) {
+            if (decl.$class === `${MetaModelNamespace}.MapDeclaration`) {
                 const processedMapDecl = this.processMapDeclaration(decl, namespace);
                 decl = processedMapDecl;
             }