Added migration testing