diff --git a/pitr/cli/internal/cmd/backup_test.go b/pitr/cli/internal/cmd/backup_test.go
index 2ddab5c5..8844c18e 100644
--- a/pitr/cli/internal/cmd/backup_test.go
+++ b/pitr/cli/internal/cmd/backup_test.go
@@ -382,18 +382,21 @@ var _ = Describe("test backup mock", func() {
 		})
 
 		It("agent server is not running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(errors.New("error")).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()
 			Expect(checkAgentServerStatus(ls)).To(BeFalse())
 		})
 
 		It("agent server are running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(nil).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()
 			Expect(checkAgentServerStatus(ls)).To(BeTrue())
 		})
 
 		It("one agent server is not running", func() {
+			mockIreq.EXPECT().Body(gomock.Any())
 			mockIreq.EXPECT().Send(gomock.Any()).Return(errors.New("failed"))
 			mockIreq.EXPECT().Send(gomock.Any()).Return(nil).AnyTimes()
 			mockIreq.EXPECT().Header(gomock.Any()).AnyTimes()