cleanup
small fix