From 5a8e8850a584756c69f94781ed0be3562bc7021e Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Fri, 25 Mar 2022 21:00:24 +0530
Subject: [PATCH] Agent: Modify schedule jobs PBA to return PostBreachData

---
 monkey/infection_monkey/post_breach/actions/schedule_jobs.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/monkey/infection_monkey/post_breach/actions/schedule_jobs.py b/monkey/infection_monkey/post_breach/actions/schedule_jobs.py
index e7845968a..7bffce8a2 100644
--- a/monkey/infection_monkey/post_breach/actions/schedule_jobs.py
+++ b/monkey/infection_monkey/post_breach/actions/schedule_jobs.py
@@ -21,5 +21,6 @@ class ScheduleJobs(PBA):
         )
 
     def run(self):
-        super(ScheduleJobs, self).run()
+        post_breach_data = super(ScheduleJobs, self).run()
         remove_scheduled_jobs()
+        return post_breach_data