diff --git a/terraform/context_input_test.go b/terraform/context_input_test.go
index 5495037162..72de23757d 100644
--- a/terraform/context_input_test.go
+++ b/terraform/context_input_test.go
@@ -767,6 +767,15 @@ func TestContext2Input_dataSourceRequiresRefresh(t *testing.T) {
 	p := testProvider("null")
 	m := testModule(t, "input-module-data-vars")
 
+	p.GetSchemaReturn = &ProviderSchema{
+		DataSources: map[string]*configschema.Block{
+			"null_data_source": {
+				Attributes: map[string]*configschema.Attribute{
+					"foo": {Type: cty.List(cty.String), Optional: true},
+				},
+			},
+		},
+	}
 	p.ReadDataDiffFn = testDataDiffFn
 
 	state := &State{