logic minor change