fixed unittests for migration changes