diff --git a/command/apply_test.go b/command/apply_test.go
index b371555f72..25a57ce8b6 100644
--- a/command/apply_test.go
+++ b/command/apply_test.go
@@ -34,6 +34,16 @@ func TestApply(t *testing.T) {
 
 	p := testProvider()
 	p.GetSchemaReturn = applyFixtureSchema()
+	p.PlanResourceChangeFn = func (req providers.PlanResourceChangeRequest) providers.PlanResourceChangeResponse {
+		return providers.PlanResourceChangeResponse{
+			PlannedState: req.ProposedNewState,
+		}
+	}
+	p.ApplyResourceChangeFn = func (req providers.ApplyResourceChangeRequest) providers.ApplyResourceChangeResponse {
+		return providers.ApplyResourceChangeResponse{
+			NewState: req.PlannedState,
+		}
+	}
 
 	ui := new(cli.MockUi)
 	c := &ApplyCommand{