Added another test case around migration