small code cleanup
small code cleanup