I think this line should be dropped? The ``ward_treatment`` is in ``store.ward_treatment`` now.