Switch from linear advection to shallow water equations