diff --git a/plugin/ui_input.go b/plugin/ui_input.go
index 8343612aa..3469e6a96 100644
--- a/plugin/ui_input.go
+++ b/plugin/ui_input.go
@@ -5,7 +5,7 @@ import (
 	"net/rpc"
 
 	"github.com/hashicorp/go-plugin"
-	"github.com/hashicorp/terraform/internal/legacy/terraform"
+	"github.com/hashicorp/terraform/terraform"
 )
 
 // UIInput is an implementation of terraform.UIInput that communicates
diff --git a/plugin/ui_input_test.go b/plugin/ui_input_test.go
index 3829e9e84..c6d7036d1 100644
--- a/plugin/ui_input_test.go
+++ b/plugin/ui_input_test.go
@@ -6,7 +6,7 @@ import (
 	"testing"
 
 	"github.com/hashicorp/go-plugin"
-	"github.com/hashicorp/terraform/internal/legacy/terraform"
+	"github.com/hashicorp/terraform/terraform"
 )
 
 func TestUIInput_impl(t *testing.T) {
diff --git a/plugin/ui_output.go b/plugin/ui_output.go
index 752c93144..c222b00cd 100644
--- a/plugin/ui_output.go
+++ b/plugin/ui_output.go
@@ -3,7 +3,7 @@ package plugin
 import (
 	"net/rpc"
 
-	"github.com/hashicorp/terraform/internal/legacy/terraform"
+	"github.com/hashicorp/terraform/terraform"
 )
 
 // UIOutput is an implementatin of terraform.UIOutput that communicates
diff --git a/plugin/ui_output_test.go b/plugin/ui_output_test.go
index 43c4ad6fc..50eadaa02 100644
--- a/plugin/ui_output_test.go
+++ b/plugin/ui_output_test.go
@@ -4,7 +4,7 @@ import (
 	"testing"
 
 	"github.com/hashicorp/go-plugin"
-	"github.com/hashicorp/terraform/internal/legacy/terraform"
+	"github.com/hashicorp/terraform/terraform"
 )
 
 func TestUIOutput_impl(t *testing.T) {