Fixed incorrect message error in the stream API

The message package is passed to the stream object and
used to evaluate correct reply message type

Change-Id: I2c9844d6447d024af1693205efd5721e2f89f22d
Signed-off-by: Vladimir Lavor <vlavor@cisco.com>

1 files changed, 4 insertions, 2 deletions

diff --git a/cmd/vpp-proxy/main.go b/cmd/vpp-proxy/main.go
index d1af5df..3c85bcf 100644
--- a/cmd/vpp-proxy/main.go
+++ b/cmd/vpp-proxy/main.go
@@ -35,8 +35,10 @@ var (
 )
 
 func init() {
-	for _, msg := range api.GetRegisteredMessages() {
-		gob.Register(msg)
+	for _, msgList := range api.GetRegisteredMessages() {
+		for _, msg := range msgList {
+			gob.Register(msg)
+		}
 	}
 }