small cleanup

small cleanup