diff --git a/go.mod b/go.mod
index 8b3cede..0f5a35d 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.23.6
 toolchain go1.24.1
 
 require (
-	github.com/compliance-framework/agent v0.1.9-0.20250708121809-c5059c3efac8
+	github.com/compliance-framework/agent v0.2.1
 	github.com/google/go-github/v71 v71.0.0
 	github.com/hashicorp/go-hclog v1.6.3
 	github.com/hashicorp/go-plugin v1.6.3
@@ -16,7 +16,7 @@ require (
 	github.com/agnivade/levenshtein v1.2.1 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
-	github.com/compliance-framework/configuration-service v0.2.12-0.20250708103936-23c66564a854 // indirect
+	github.com/compliance-framework/api v0.4.0 // indirect
 	github.com/defenseunicorns/go-oscal v0.6.2 // indirect
 	github.com/fatih/color v1.18.0 // indirect
 	github.com/go-ini/ini v1.67.0 // indirect
diff --git a/go.sum b/go.sum
index 3cdc3a7..8d4086b 100644
--- a/go.sum
+++ b/go.sum
@@ -26,10 +26,10 @@ github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK3
 github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
-github.com/compliance-framework/agent v0.1.9-0.20250708121809-c5059c3efac8 h1:KESs41QePWTLYZMgKAWuQUBB3VMn9T774nI8+aeJF/Y=
-github.com/compliance-framework/agent v0.1.9-0.20250708121809-c5059c3efac8/go.mod h1:K9tKkawU6Bp0xmtcCsHRL29tj0eJQRtK4lFbq7ALzKs=
-github.com/compliance-framework/configuration-service v0.2.12-0.20250708103936-23c66564a854 h1:SJnaFykNrKhH1SpJhufQSghI0EmtF2P89K9hyFsLpqo=
-github.com/compliance-framework/configuration-service v0.2.12-0.20250708103936-23c66564a854/go.mod h1:iLh9bNKJHRIk2OrfpcoEz3VuX+rXyo7PA3Qh/pxeWLg=
+github.com/compliance-framework/agent v0.2.1 h1:I2cvHRdkBiIXeud7STptpg0+pzHBSUMiFxIuI4EzdGc=
+github.com/compliance-framework/agent v0.2.1/go.mod h1:fpUMZejzNNfwadGnrN8HpAAyka+UANx8LVhiLZeoPhg=
+github.com/compliance-framework/api v0.4.0 h1:wMptuEwrHfUykAmDuXIAc5VeLgVBmZUHJH1ASPUkLaY=
+github.com/compliance-framework/api v0.4.0/go.mod h1:ViBBOdBQ5dG/AdBGPIzEiJ0vuPDyChogBPVE2VfKwKY=
 github.com/containerd/log v0.1.0 h1:TCJt7ioM2cr/tfR8GPbGf9/VRAX8D2B4PjzCpfX540I=
 github.com/containerd/log v0.1.0/go.mod h1:VRRf09a7mHDIRezVKTRCrOq78v577GXq3bSa3EhrzVo=
 github.com/containerd/platforms v0.2.1 h1:zvwtM3rz2YHPQsF2CHYM8+KtB5dvhISiXh5ZpSBQv6A=