diff --git a/pkg/mapper/examples/retry/Makefile b/pkg/mapper/examples/retry/Makefile
index c75d4910..96700aa7 100644
--- a/pkg/mapper/examples/retry/Makefile
+++ b/pkg/mapper/examples/retry/Makefile
@@ -4,7 +4,7 @@ build:
 
 .PHONY: image
 image: build
-	docker build -t "quay.io/numaio/numaflow-go/map-retry:v0.6.0" --target retry .
+	docker build -t "quay.io/numaio/numaflow-go/map-retry:v0.6.1" --target retry .
 
 clean:
 	-rm -rf ./dist
diff --git a/pkg/mapper/examples/retry/go.mod b/pkg/mapper/examples/retry/go.mod
index 38ebb5c6..2be14290 100644
--- a/pkg/mapper/examples/retry/go.mod
+++ b/pkg/mapper/examples/retry/go.mod
@@ -2,7 +2,7 @@ module retry
 
 go 1.20
 
-require github.com/numaproj/numaflow-go v0.6.0
+require github.com/numaproj/numaflow-go v0.6.1-0.20240212202512-715955f1e069
 
 require (
 	github.com/golang/protobuf v1.5.3 // indirect
diff --git a/pkg/mapper/examples/retry/go.sum b/pkg/mapper/examples/retry/go.sum
index 8b1ed758..d5a279ec 100644
--- a/pkg/mapper/examples/retry/go.sum
+++ b/pkg/mapper/examples/retry/go.sum
@@ -6,6 +6,8 @@ github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/
 github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
 github.com/numaproj/numaflow-go v0.6.0 h1:gqTX1u1pFJJhX/3l3zYM8aLqRSHEainYrgBIollL0js=
 github.com/numaproj/numaflow-go v0.6.0/go.mod h1:5zwvvREIbqaCPCKsNE1MVjVToD0kvkCh2Z90Izlhw5U=
+github.com/numaproj/numaflow-go v0.6.1-0.20240212202512-715955f1e069 h1:LBMIrUytoxuZazEnwxo+WPAxjHL1rrVhphTyByLiDYY=
+github.com/numaproj/numaflow-go v0.6.1-0.20240212202512-715955f1e069/go.mod h1:WoMt31+h3up202zTRI8c/qe42B8UbvwLe2mJH0MAlhI=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
 golang.org/x/net v0.9.0 h1:aWJ/m6xSmxWBx+V0XRHTlrYrPG56jKsLdTFmsSsCzOM=