Fix bug on UI

@cgvdw het hierdie bygesit