From aedcbc9362ca1dea810745d116ddb5f16794c163 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ola=20Holmstr=C3=B6m?= <ola.holmstrom@gmail.com>
Date: Wed, 13 May 2015 23:17:16 +0200
Subject: [PATCH] fix documentation

---
 melody.go  | 2 +-
 session.go | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/melody.go b/melody.go
index ac45bc4..0b4b982 100644
--- a/melody.go
+++ b/melody.go
@@ -58,7 +58,7 @@ func (m *Melody) HandleMessage(fn func(*Session, []byte)) {
 }
 
 // Fires when a session has an error.
-func (m *Melody) HandleError(fn handleErrorFunc) {
+func (m *Melody) HandleError(fn func(*Session, error)) {
 	m.errorHandler = fn
 }
 
diff --git a/session.go b/session.go
index fb7eeb4..7090ac8 100644
--- a/session.go
+++ b/session.go
@@ -83,12 +83,12 @@ func (s *Session) readPump(messageHandler handleMessageFunc, errorHandler handle
 	}
 }
 
-// Write a message to session.
+// Write message to session.
 func (s *Session) Write(msg []byte) {
 	s.writeMessage(&envelope{t: websocket.TextMessage, msg: msg})
 }
 
-// Close a session.
+// Close session.
 func (s *Session) Close() {
 	s.writeMessage(&envelope{t: websocket.CloseMessage, msg: []byte{}})
 }