Fixed StokesFlow to work as well