diff --git a/terraform/context_apply_test.go b/terraform/context_apply_test.go
index 089660a81b..04af6aa55a 100644
--- a/terraform/context_apply_test.go
+++ b/terraform/context_apply_test.go
@@ -5782,6 +5782,7 @@ func TestContext2Apply_outputDiffVars(t *testing.T) {
 
 		result := s.MergeDiff(d)
 		result.ID = "foo"
+		result.Attributes["foo"] = "bar"
 		return result, nil
 	}
 	p.DiffFn = func(*InstanceInfo, *InstanceState, *ResourceConfig) (*InstanceDiff, error) {