Dang, I know it rushed because LAW.
But, the author could make the ending a normal one and not JUST RESET THE FUKIN MANGA.
Just give casual He fell into the pool > then he swim upward to the surface > then he see face to face with the girl > And the girl just said something like Welcome Back.
Even tho it's not perfect, at least it's a warm ending.