Fix button a bit