Actually use the right context type.