Updates to Migration model for rerun on Neon