diff --git a/states/resource_test.go b/states/resource_test.go
index 0ce422002..ffe81ee7b 100644
--- a/states/resource_test.go
+++ b/states/resource_test.go
@@ -15,7 +15,7 @@ func TestResourceInstanceDeposeCurrentObject(t *testing.T) {
 	var dk DeposedKey
 
 	t.Run("first depose", func(t *testing.T) {
-		dk = is.deposeCurrentObject() // dk is randomly-generated but should be eight characters long
+		dk = is.deposeCurrentObject(NotDeposed) // dk is randomly-generated but should be eight characters long
 		t.Logf("deposedKey is %q", dk)
 
 		if got := is.Current; got != nil {
@@ -33,7 +33,7 @@ func TestResourceInstanceDeposeCurrentObject(t *testing.T) {
 	})
 
 	t.Run("second depose", func(t *testing.T) {
-		notDK := is.deposeCurrentObject()
+		notDK := is.deposeCurrentObject(NotDeposed)
 		if notDK != NotDeposed {
 			t.Errorf("got deposedKey %q; want NotDeposed", notDK)
 		}