little code clean up at the end of the day