Small UI fix in reference to #32