diff --git a/lua/cmp/core.lua b/lua/cmp/core.lua
index c4e6a14..e5d2b10 100644
--- a/lua/cmp/core.lua
+++ b/lua/cmp/core.lua
@@ -74,7 +74,11 @@ end
 core.on_keymap = function(keys, fallback)
   for key, action in pairs(config.get().mapping) do
     if key == keys then
-      return action.invoke(core, fallback)
+      if type(action) == 'function' then
+        return action(core, fallback)
+      else
+        return action.invoke(core, fallback)
+      end
     end
   end