Smart partition recovery