From 7c0012d355d25ce61c90a119e914074e755f404b Mon Sep 17 00:00:00 2001
From: Luis Aleixo <luis.aleixo@cern.ch>
Date: Thu, 27 Jan 2022 17:23:42 +0100
Subject: [PATCH] handled case when back navigation is performed with changing
 the short range option

---
 cara/apps/calculator/model_generator.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/cara/apps/calculator/model_generator.py b/cara/apps/calculator/model_generator.py
index 7e3180b1..893144a4 100644
--- a/cara/apps/calculator/model_generator.py
+++ b/cara/apps/calculator/model_generator.py
@@ -418,6 +418,13 @@ class FormData:
 
         infected_occupants = self.infected_people
 
+        if self.short_range_option == "short_range_no":
+            sr_presence=[]
+            sr_activities=[]
+        else:
+            sr_presence=self.short_range_intervals()
+            sr_activities=self.short_range_activities()
+
         infected = mc.InfectedPopulation(
             number=infected_occupants,
             virus=virus,
@@ -426,8 +433,8 @@ class FormData:
             activity=activity,
             expiration=expiration,
             host_immunity=0.,
-            short_range_presence=self.short_range_intervals(),
-            short_range_activities=self.short_range_activities(),
+            short_range_presence=sr_presence,
+            short_range_activities=sr_activities,
         )
         return infected