diff --git a/src/WebSocketsClient.h b/src/WebSocketsClient.h
index 6a2c60e..589a0d7 100644
--- a/src/WebSocketsClient.h
+++ b/src/WebSocketsClient.h
@@ -31,7 +31,7 @@
 class WebSocketsClient: private WebSockets {
     public:
 
-        typedef void (*WebSocketClientEvent)(WStype_t type, uint8_t * payload, size_t length);
+        typedef std::function<void (WStype_t type, uint8_t * payload, size_t length)> WebSocketClientEvent;
 
         WebSocketsClient(void);
         ~WebSocketsClient(void);
diff --git a/src/WebSocketsServer.h b/src/WebSocketsServer.h
index ebc76b2..e7c00e6 100644
--- a/src/WebSocketsServer.h
+++ b/src/WebSocketsServer.h
@@ -36,7 +36,7 @@
 class WebSocketsServer: private WebSockets {
 public:
 
-        typedef void (*WebSocketServerEvent)(uint8_t num, WStype_t type, uint8_t * payload, size_t length);
+        typedef std::function<void (uint8_t num, WStype_t type, uint8_t * payload, size_t length)> WebSocketServerEvent;
 
         WebSocketsServer(uint16_t port);
         ~WebSocketsServer(void);