FIX: Return original button