diff --git a/pkg/batchmapper/service_test.go b/pkg/batchmapper/service_test.go
index f2f4cccd..55034a5b 100644
--- a/pkg/batchmapper/service_test.go
+++ b/pkg/batchmapper/service_test.go
@@ -257,6 +257,13 @@ func TestService_MapFnStream(t *testing.T) {
 
 			wg.Add(1)
 			go func() {
+				defer func() {
+					f := recover()
+					if f != nil {
+						err = fmt.Errorf("got a panic")
+						wg.Done()
+					}
+				}()
 				defer wg.Done()
 				err = fs.BatchMapFn(udfBatchMapFnStream)
 				close(outputCh)