diff --git a/sdks/go/pkg/beam/io/natsio/write.go b/sdks/go/pkg/beam/io/natsio/write.go
index 2da3dc35127c..8991ef8cac16 100644
--- a/sdks/go/pkg/beam/io/natsio/write.go
+++ b/sdks/go/pkg/beam/io/natsio/write.go
@@ -38,7 +38,7 @@ func init() {
 type ProducerMessage struct {
 	Subject string
 	ID      string
-	Headers nats.Header
+	Headers map[string][]string
 	Data    []byte
 }
 
diff --git a/sdks/go/pkg/beam/io/natsio/write_test.go b/sdks/go/pkg/beam/io/natsio/write_test.go
index de5ca06188f1..5e9387ece5f6 100644
--- a/sdks/go/pkg/beam/io/natsio/write_test.go
+++ b/sdks/go/pkg/beam/io/natsio/write_test.go
@@ -156,7 +156,7 @@ func TestWrite(t *testing.T) {
 				ProducerMessage{
 					Subject: subject,
 					ID:      "1",
-					Headers: nats.Header{"key": []string{"val"}},
+					Headers: map[string][]string{"key": {"val"}},
 					Data:    []byte("msg1"),
 				},
 			},