diff --git a/terraform/context_test.go b/terraform/context_test.go
index e4359685f7..5ee9a8c71b 100644
--- a/terraform/context_test.go
+++ b/terraform/context_test.go
@@ -4454,6 +4454,9 @@ func TestContext2Apply_hook(t *testing.T) {
 	if !h.PostApplyCalled {
 		t.Fatal("should be called")
 	}
+	if !h.PostStateUpdateCalled {
+		t.Fatalf("should call post state update")
+	}
 }
 
 func TestContext2Apply_idAttr(t *testing.T) {