From 2c543a43d430115dc4f34be585a69ceae1837ca3 Mon Sep 17 00:00:00 2001
From: Srini Venkatramanan <srini@virginia.edu>
Date: Thu, 3 Sep 2020 18:31:35 -0400
Subject: [PATCH] Update patchsim.py

---
 patchsim.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/patchsim.py b/patchsim.py
index fc67ccf..cc07623 100644
--- a/patchsim.py
+++ b/patchsim.py
@@ -262,7 +262,7 @@ def load_vax(configs, params, patch_df):
     )
     vax_delay = int(configs.get("VaxDelay", 0))
     
-    vax_df = vax_df[vax_df.Day<params["T"]] ### Skipping vaxs after end of simulation
+    vax_df = vax_df[vax_df.Day<params["T"] - vax_delay] ### Skipping vaxs which get applied after end of simulation
     patch_idx = {id_: i for i, id_ in enumerate(patch_df["id"])}
     for day, id_, count in vax_df.itertuples(index=False, name=None):
         idx = patch_idx[id_]