diff --git a/_examples/websocket-initfunc/server/server.go b/_examples/websocket-initfunc/server/server.go
index 7979800fe23..8e0f7946bc5 100644
--- a/_examples/websocket-initfunc/server/server.go
+++ b/_examples/websocket-initfunc/server/server.go
@@ -63,7 +63,7 @@ func main() {
 				return true
 			},
 		},
-		InitFunc: func(ctx context.Context, initPayload transport.InitPayload) (*transport.InitPayload, context.Context, error) {
+		InitFunc: func(ctx context.Context, initPayload transport.InitPayload) (context.Context, *transport.InitPayload, error) {
 			return webSocketInit(ctx, initPayload)
 		},
 	})