diff --git a/internal/cmd/backup.go b/internal/cmd/backup.go
index 69d58132..5b063eec 100644
--- a/internal/cmd/backup.go
+++ b/internal/cmd/backup.go
@@ -359,7 +359,8 @@ func restoreCmdFunc(cmd *cobra.Command, args []string) error {
 			if err := relationshipWriter.Send(&v1.BulkImportRelationshipsRequest{
 				Relationships: batch,
 			}); err != nil {
-				return fmt.Errorf("error sending batch to server: %w", err)
+				_, closeErr := relationshipWriter.CloseAndRecv()
+				return fmt.Errorf("error sending batch to server: %w", errors.Join(err, closeErr))
 			}
 
 			// Reset the relationships in the batch