UI fix #2