diff --git a/lib/services/saml_test.go b/lib/services/saml_test.go
index 832fe1fcdc8b2..0689a8e6457d6 100644
--- a/lib/services/saml_test.go
+++ b/lib/services/saml_test.go
@@ -24,7 +24,7 @@ import (
 	"github.com/gravitational/teleport/lib/utils"
 
 	"gopkg.in/check.v1"
-	kyaml "k8s.io/apimachinery-go/pkg/util/yaml"
+	kyaml "k8s.io/apimachinery/pkg/util/yaml"
 )
 
 type SAMLSuite struct{}