Fix live reload
cleanup