Correcting callback for new easy_edit feature set