diff --git a/lib/typeprof/core/service.rb b/lib/typeprof/core/service.rb
index c9f80d5a4..7600095cf 100644
--- a/lib/typeprof/core/service.rb
+++ b/lib/typeprof/core/service.rb
@@ -176,7 +176,8 @@ def definitions(path, pos)
               site.resolve(genv, nil) do |me, _ty, _mid, _orig_ty|
                 next unless me
                 me.defs.each do |mdef|
-                  defs << [mdef.node.lenv.path, mdef.node.mid_code_range]
+                  code_range = mdef.node.respond_to?(:mid_code_range) ? mdef.node.mid_code_range : mdef.node.code_range
+                  defs << [mdef.node.lenv.path, code_range]
                 end
               end
             end