a small clean up!