diff --git a/terraform/context_apply_test.go b/terraform/context_apply_test.go
index 87a994bf81..2b40d6b86f 100644
--- a/terraform/context_apply_test.go
+++ b/terraform/context_apply_test.go
@@ -11928,8 +11928,8 @@ resource "test_resource" "foo" {
 	})
 
 	p := testProvider("test")
-	p.ApplyFn = testApplyFn
-	p.DiffFn = testDiffFn
+	p.ApplyResourceChangeFn = testApplyFn
+	p.PlanResourceChangeFn = testDiffFn
 
 	ctx := testContext2(t, &ContextOpts{
 		Config: m,