cleanup
Small cleanup