small clean up from previous commit