Added one unit test, repaired migration