minor change in TODO