diff --git a/src/augmentation.ts b/src/augmentation.ts
index 065176368..cc32187ec 100644
--- a/src/augmentation.ts
+++ b/src/augmentation.ts
@@ -1,6 +1,21 @@
+import Math from './math';
+import MathCommand from './mathcommand';
+import MathEditing from './mathediting';
+import MathUI from './mathui';
 import type { KatexOptions } from './typings-external';
 
 declare module '@ckeditor/ckeditor5-core' {
+
+	interface PluginsMap {
+		[ Math.pluginName ]: Math;
+		[ MathEditing.pluginName ]: MathEditing;
+		[ MathUI.pluginName ]: MathUI;
+	}
+
+	interface CommandsMap {
+		math: MathCommand;
+	}
+
 	interface EditorConfig {
 		math?: {
 			engine?:
diff --git a/src/index.ts b/src/index.ts
index 433ed384f..39dc1a626 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -4,3 +4,5 @@
 
 export { default as Math } from './math';
 export { default as AutoformatMath } from './autoformatmath';
+
+import "./augmentation.js";