Correct migration script