Add missing context parameter
Merge pull request #353 from bcyrill/cp_patch3

Add missing context parameter