diff --git a/packages/cli/src/templates/context.ejs b/packages/cli/src/templates/context.ejs
index ae02ec0..e70eb97 100644
--- a/packages/cli/src/templates/context.ejs
+++ b/packages/cli/src/templates/context.ejs
@@ -1,8 +1,8 @@
-import { ContextDefinition } from "jsonld";
+import { LdoJsonldContext } from "@ldo/jsonld-dataset-proxy";
 
 /**
  * =============================================================================
  * <%- fileName %>Context: JSONLD Context for <%- fileName %>
  * =============================================================================
  */
-export const <%- fileName %>Context: ContextDefinition = <%- context %>;
+export const <%- fileName %>Context: LdoJsonldContext = <%- context %>;
diff --git a/packages/schema-converter-shex/src/context/JsonLdContextBuilder.ts b/packages/schema-converter-shex/src/context/JsonLdContextBuilder.ts
index fe28d73..1c67445 100644
--- a/packages/schema-converter-shex/src/context/JsonLdContextBuilder.ts
+++ b/packages/schema-converter-shex/src/context/JsonLdContextBuilder.ts
@@ -1,5 +1,6 @@
 import type { Annotation } from "shexj";
-import type { ContextDefinition, ExpandedTermDefinition } from "jsonld";
+import type { ExpandedTermDefinition } from "jsonld";
+import type { LdoJsonldContext } from "@ldo/jsonld-dataset-proxy";
 
 /**
  * Name functions
@@ -184,8 +185,8 @@ export class JsonLdContextBuilder {
     }
   }
 
-  generateJsonldContext(): ContextDefinition {
-    const contextDefnition: ContextDefinition = {};
+  generateJsonldContext(): LdoJsonldContext {
+    const contextDefnition: LdoJsonldContext = {};
     const namesMap = this.generateNames();
     Object.entries(namesMap).forEach(([iri, name]) => {
       if (this.iriTypes[iri]) {