clean up

small code clean up