Small code clean up.