Simple code clean up.