diff --git a/terraform/context_apply_test.go b/terraform/context_apply_test.go
index ba0e98694..b1b70283c 100644
--- a/terraform/context_apply_test.go
+++ b/terraform/context_apply_test.go
@@ -8281,7 +8281,7 @@ func TestContext2Apply_unknownAttribute(t *testing.T) {
 
 	state, diags := ctx.Apply()
 	if !diags.HasErrors() {
-		t.Fatal("should error with UnknownVariableValue")
+		t.Error("should error, because attribute 'unknown' is still unknown after apply")
 	}
 
 	actual := strings.TrimSpace(state.String())
diff --git a/terraform/context_test.go b/terraform/context_test.go
index 78a724915..4b4c28091 100644
--- a/terraform/context_test.go
+++ b/terraform/context_test.go
@@ -531,6 +531,12 @@ func testProviderSchema(name string) *ProviderSchema {
 						Type:     cty.String,
 						Computed: true,
 					},
+
+					// Generated by testDiffFn if compute = "unknown" is set in the test config
+					"unknown": {
+						Type:     cty.String,
+						Computed: true,
+					},
 				},
 			},
 			name + "_eip": {