Add missing method

It was forgotten during the merge