Add the model's context to custom type function. Note: Test only passes when anchor starts supporting contexts.