Small cleanup
Small cleanup