Added missing event handler