Fixed the UI buttons