UI bugfix. Closes #308