Fix rollback wait parameter bug