Add unit test to prove migration