I think that Takae can pass away sooner, because for Keisuke the realisation and coming to terms with Takae's wish were kind of abrupt and not really wanted, so those can be the same for Mai too, if the author, of course, didn't want to add even more symbolism with the wedding.