If they do a sequel, I’d be fine with them using the exact same map for the school proper, but setting it during a different period and updating the world around it. Set it during WWI and show us what the wizards were doing when the rest of the world burnt down, set it in the 1970’s and show how they coped with the muggles rapidly advancing technology. I’d suggest to have it set during the modern day after the events of the books/movies to show how they responded to smartphones and the internet but with how they portrayed the end of the 19th century, a modern day setting would be horrifying, so I’d rather it be kept as a period piece.