migration check is removed since it is unneccessary because we are already migrating partition loaded flag as true.Also has some test improvements