logic minor issue fix