Add missing context parameter