a little fix

worked well but seems not right logically