Add missing handler