diff --git a/internal/lsp/source/highlight.go b/internal/lsp/source/highlight.go
index f5c28c2..849bc80 100644
--- a/internal/lsp/source/highlight.go
+++ b/internal/lsp/source/highlight.go
@@ -91,6 +91,10 @@
 			inReturnList = inReturnList || path[0] != returnStmt
 		}
 	}
+	// Cursor is not in a function.
+	if enclosingFunc == nil {
+		return nil, nil
+	}
 	// If the cursor is on a "return" or "func" keyword, we should highlight all of the exit
 	// points of the function, including the "return" and "func" keywords.
 	highlightAllReturnsAndFunc := path[0] == returnStmt || path[0] == enclosingFunc