Fix bug for rollback migration