Fix hot reloading