1 files changed, 9 insertions, 9 deletions

diff --git a/pkg/messages/types/svcEntityMessage.go b/pkg/messages/types/svcEntityMessage.go
index b955da5..87d7e1b 100644
--- a/pkg/messages/types/svcEntityMessage.go
+++ b/pkg/messages/types/svcEntityMessage.go
@@ -2,25 +2,25 @@ package messages
 
 import (
         "github.com/pektezol/bitreader"
-        "github.com/pektezol/sdp.go/pkg/writer"
+        "github.com/pektezol/sdp.go/pkg/types"
 )
 
 type SvcEntityMessage struct {
-        EntityIndex uint16
-        ClassId     uint16
-        Length      uint16
-        Data        []byte
+        EntityIndex uint16 `json:"entity_index"`
+        ClassId     uint16 `json:"class_id"`
+        Length      uint16 `json:"length"`
+        Data        []byte `json:"data"`
 }
 
-func ParseSvcEntityMessage(reader *bitreader.Reader) SvcEntityMessage {
+func ParseSvcEntityMessage(reader *bitreader.Reader, demo *types.Demo) SvcEntityMessage {
         svcEntityMessage := SvcEntityMessage{
                 EntityIndex: uint16(reader.TryReadBits(11)),
                 ClassId:     uint16(reader.TryReadBits(9)),
                 Length:      uint16(reader.TryReadBits(11)),
         }
         svcEntityMessage.Data = reader.TryReadBitsToSlice(uint64(svcEntityMessage.Length))
-        writer.TempAppendLine("\t\tEntity Index: %d", svcEntityMessage.EntityIndex)
-        writer.TempAppendLine("\t\tClass ID: %d", svcEntityMessage.ClassId)
-        writer.TempAppendLine("\t\tData: %v", svcEntityMessage.Data)
+        demo.Writer.TempAppendLine("\t\tEntity Index: %d", svcEntityMessage.EntityIndex)
+        demo.Writer.TempAppendLine("\t\tClass ID: %d", svcEntityMessage.ClassId)
+        demo.Writer.TempAppendLine("\t\tData: %v", svcEntityMessage.Data)
         return svcEntityMessage
 }