diff --git a/server.go b/server.go
index 2b7903fb6..7186e55cc 100644
--- a/server.go
+++ b/server.go
@@ -220,6 +220,7 @@ func NewServer(opts ...ServerOption) (*Server, error) {
 
 		logger: NopLogger,
 	}
+	s.diagnostics.server = s
 
 	for _, opt := range opts {
 		err := opt(s)