Minor button ui fix...