Add missing event handler