little code clean up