add some logic.

fyi.