diff --git a/pkg/policy/policy_test.go b/pkg/policy/policy_test.go
index 3740d36..bf389fb 100644
--- a/pkg/policy/policy_test.go
+++ b/pkg/policy/policy_test.go
@@ -101,6 +101,7 @@ func TestRegoEvaluator_Evaluate(t *testing.T) {
 
 			if tc.expectSuccess {
 				assert.NoErrorf(t, err, "Evaluate failed")
+				assert.True(t, rs.Allowed(), "Evaluate should have succeeded")
 			} else {
 				assert.False(t, rs.Allowed(), "Evaluate should have failed")
 			}
diff --git a/pkg/policy/rego.go b/pkg/policy/rego.go
index 48ca0f3..c6851b0 100644
--- a/pkg/policy/rego.go
+++ b/pkg/policy/rego.go
@@ -29,7 +29,7 @@ type regoEvaluator struct {
 func NewRegoEvaluator(debug bool) PolicyEvaluator {
 	return &regoEvaluator{
 		debug: debug,
-		query: "data.attestations.allow",
+		query: "data.attest.allow",
 	}
 }
 
diff --git a/pkg/policy/testdata/mock-tuf-allow/doi/policy.rego b/pkg/policy/testdata/mock-tuf-allow/doi/policy.rego
index 5e3b8ab..b1b188a 100644
--- a/pkg/policy/testdata/mock-tuf-allow/doi/policy.rego
+++ b/pkg/policy/testdata/mock-tuf-allow/doi/policy.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 
 import rego.v1
 
diff --git a/pkg/policy/testdata/mock-tuf-deny/doi/policy.rego b/pkg/policy/testdata/mock-tuf-deny/doi/policy.rego
index ad73140..5e4ac7e 100644
--- a/pkg/policy/testdata/mock-tuf-deny/doi/policy.rego
+++ b/pkg/policy/testdata/mock-tuf-deny/doi/policy.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 
 import rego.v1
 
diff --git a/pkg/policy/testdata/mock-tuf-verify-sig/doi/policy.rego b/pkg/policy/testdata/mock-tuf-verify-sig/doi/policy.rego
index e2456af..8952fb1 100644
--- a/pkg/policy/testdata/mock-tuf-verify-sig/doi/policy.rego
+++ b/pkg/policy/testdata/mock-tuf-verify-sig/doi/policy.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 
 import rego.v1
 
diff --git a/pkg/policy/testdata/mock-tuf-wrong-key/doi/policy.rego b/pkg/policy/testdata/mock-tuf-wrong-key/doi/policy.rego
index 131129a..aaa78d7 100644
--- a/pkg/policy/testdata/mock-tuf-wrong-key/doi/policy.rego
+++ b/pkg/policy/testdata/mock-tuf-wrong-key/doi/policy.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 
 import rego.v1
 
diff --git a/test/testdata/local-policy/doi/policy.rego b/test/testdata/local-policy/doi/policy.rego
index ed35cbe..d4ec593 100644
--- a/test/testdata/local-policy/doi/policy.rego
+++ b/test/testdata/local-policy/doi/policy.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 
 import rego.v1
 
diff --git a/test/testdata/local-policy/doi/policy_test.rego b/test/testdata/local-policy/doi/policy_test.rego
index fa81aa9..e2bff5c 100644
--- a/test/testdata/local-policy/doi/policy_test.rego
+++ b/test/testdata/local-policy/doi/policy_test.rego
@@ -1,4 +1,4 @@
-package docker
+package attest
 import rego.v1
 
 config := {"keys": []}