diff --git a/terraform/context.go b/terraform/context.go
index 702685399..eebae1c15 100644
--- a/terraform/context.go
+++ b/terraform/context.go
@@ -372,7 +372,9 @@ func (c *Context) Apply() (*State, error) {
 		graph, err = c.Graph(&ContextGraphOpts{Validate: true})
 	} else {
 		graph, err = (&ApplyGraphBuilder{
+			Module:       c.module,
 			Diff:         c.diff,
+			State:        c.state,
 			Providers:    c.components.ResourceProviders(),
 			Provisioners: c.components.ResourceProvisioners(),
 		}).Build(RootModulePath)