minor fix on buttons