diff --git a/flow/activities/flowable.go b/flow/activities/flowable.go
index 885ad38c38..4e003930f8 100644
--- a/flow/activities/flowable.go
+++ b/flow/activities/flowable.go
@@ -338,7 +338,7 @@ func (a *FlowableActivity) StartFlow(ctx context.Context,
 		log.Warnf("failed to push records: %v", err)
 		return nil, fmt.Errorf("failed to push records: %w", err)
 	}
-
+	done <- struct{}{}
 	err = errGroup.Wait()
 	if err != nil {
 		return nil, fmt.Errorf("failed to pull records: %w", err)
@@ -386,7 +386,6 @@ func (a *FlowableActivity) StartFlow(ctx context.Context,
 
 	pushedRecordsWithCount := fmt.Sprintf("pushed %d records", numRecords)
 	activity.RecordHeartbeat(ctx, pushedRecordsWithCount)
-	done <- struct{}{}
 
 	return res, nil
 }