Fix up migrations