Fix velocity field stuff after changing API