diff --git a/lua/blink-cmp-rg/init.lua b/lua/blink-cmp-rg/init.lua
index 373fea3..2897fbe 100644
--- a/lua/blink-cmp-rg/init.lua
+++ b/lua/blink-cmp-rg/init.lua
@@ -2,9 +2,13 @@
 
 ---@class blink-cmp-rg.Options
 ---@field prefix_min_len? number
----@field get_command? fun(self: blink.cmp.Source, prefix: string): string[]
----@field get_prefix? fun(self: blink.cmp.Source): string
+---@field get_command? fun(self: blink.cmp.Context, prefix: string): string[]
+---@field get_prefix? fun(self: blink.cmp.Context): string
 
+---@class blink-cmp-rg.RgSource : blink.cmp.Source
+---@field prefix_min_len number
+---@field get_command fun(self: blink.cmp.Context, prefix: string): string[]
+---@field get_prefix fun(self: blink.cmp.Context): string
 local RgSource = {}
 
 ---@param opts blink-cmp-rg.Options