small fix, clean up