Fixing a button bug

Signed-off-by: mossrocket <peter@mossrocket.com>