small change.
small code cleanup