cleanup
small tweak