Updated migration so that rollback to the beginning works.