clean up
small fix