diff --git a/go.mod b/go.mod
index d057046..39ffae4 100644
--- a/go.mod
+++ b/go.mod
@@ -4,7 +4,7 @@ go 1.23.2
 
 require (
 	github.com/chris-cmsoft/conftojson v0.0.0-20241105132434-7c8873e5cbb6
-	github.com/compliance-framework/agent v0.1.9-0.20250708121809-c5059c3efac8
+	github.com/compliance-framework/agent v0.2.1
 	github.com/hashicorp/go-hclog v1.6.3
 	github.com/hashicorp/go-plugin v1.6.2
 )
@@ -14,7 +14,7 @@ require (
 	github.com/agnivade/levenshtein v1.2.0 // 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 f0f578e..e9b2a94 100644
--- a/go.sum
+++ b/go.sum
@@ -32,10 +32,10 @@ github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UF
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
 github.com/chris-cmsoft/conftojson v0.0.0-20241105132434-7c8873e5cbb6 h1:50LCDU9IpLMdBC9+6+g7EReo50gr0BBFevGRaLwmdJQ=
 github.com/chris-cmsoft/conftojson v0.0.0-20241105132434-7c8873e5cbb6/go.mod h1:lGxtdIwiDsj1U+vUxAydlacBWodyOoBjgJ9IX6T8YTQ=
-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=