Little ui fix on edit round