fixing post upgrade code