small clean up in code