Small UI fix

I think this fixes that bug: