diff --git a/go.mod b/go.mod
index aeb95a3..24bb3a2 100644
--- a/go.mod
+++ b/go.mod
@@ -6,7 +6,7 @@ require (
 	github.com/Azure/azure-sdk-for-go/sdk/azidentity v1.8.2
 	github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/compute/armcompute v1.0.0
 	github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork v1.0.0
-	github.com/compliance-framework/agent v0.2.0
+	github.com/compliance-framework/agent v0.2.1
 	github.com/hashicorp/go-hclog v1.5.0
 	github.com/hashicorp/go-plugin v1.6.2
 )
@@ -19,7 +19,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.15.0 // indirect
 	github.com/go-ini/ini v1.67.0 // indirect
diff --git a/go.sum b/go.sum
index df40f11..a3405b4 100644
--- a/go.sum
+++ b/go.sum
@@ -50,10 +50,10 @@ github.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=
 github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
 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.2.0 h1:cnX0w/HLFpCwxIQWzPBMiqvfsqDrRT5K3umPDQPLO30=
-github.com/compliance-framework/agent v0.2.0/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=