Fix button positioning