Fix W bal plotting by distance

.. since we only smoothed time

Fixes #803