diff --git a/pkg/api/kpm_pkg_test.go b/pkg/api/kpm_pkg_test.go
index fe37dbb7..f77cd6a8 100644
--- a/pkg/api/kpm_pkg_test.go
+++ b/pkg/api/kpm_pkg_test.go
@@ -37,7 +37,7 @@ func TestPackageApi(t *testing.T) {
 	schemas, err := pkg.GetAllSchemaTypeMapping()
 	assert.Equal(t, err, nil)
 	assert.Equal(t, len(schemas), 3)
-	assert.Equal(t, len(schemas["."]), 2)
+	assert.Equal(t, len(schemas["."]), 4)
 	assert.Equal(t, len(schemas[filepath.Join("sub")]), 1)
 	assert.Equal(t, len(schemas[filepath.Join("sub", "sub1")]), 2)