diff --git a/README.md b/README.md
index b3fc3d8..40eb0c9 100644
--- a/README.md
+++ b/README.md
@@ -68,7 +68,9 @@ func main() {
 	if err != nil {
 		log.Fatal(err)
 	}
-	log.Fatal(srv.Run(context.Background(), &mcp.StdioTransport{}))
+	if err := srv.Run(context.Background(), &mcp.StdioTransport{}); err != nil {
+		log.Fatal(err)
+	}
 }
 ```
 
diff --git a/docs/docs/tutorials/first-server.md b/docs/docs/tutorials/first-server.md
index 39bcd16..ba6c4ed 100644
--- a/docs/docs/tutorials/first-server.md
+++ b/docs/docs/tutorials/first-server.md
@@ -84,7 +84,9 @@ func main() {
 	if err != nil {
 		log.Fatal(err)
 	}
-	log.Fatal(srv.Run(context.Background(), &mcp.StdioTransport{}))
+	if err := srv.Run(context.Background(), &mcp.StdioTransport{}); err != nil {
+		log.Fatal(err)
+	}
 }
 ```