Fixed minor issue with button display