small code clean up.