Add specs for migration