minor ui improvement