Add missing implementation