diff --git a/hack/.import-aliases b/hack/.import-aliases
index 78c2475f8a0..56b7d413f57 100644
--- a/hack/.import-aliases
+++ b/hack/.import-aliases
@@ -32,5 +32,6 @@
   "k8s.io/api/scheduling/v1beta1": "schedulingv1beta1",
   "k8s.io/api/settings/v1alpha1": "settingsv1alpha1",
   "k8s.io/api/storage/v1": "storagev1",
-  "k8s.io/api/storage/v1alpha1": "storagev1alpha1"
+  "k8s.io/api/storage/v1alpha1": "storagev1alpha1",
+  "k8s.io/api/storage/v1beta1": "storagev1beta1"
 }
\ No newline at end of file
diff --git a/test/e2e/storage/volume_provisioning.go b/test/e2e/storage/volume_provisioning.go
index 407dfa31abc..ff873ea27f3 100644
--- a/test/e2e/storage/volume_provisioning.go
+++ b/test/e2e/storage/volume_provisioning.go
@@ -31,7 +31,7 @@ import (
 	v1 "k8s.io/api/core/v1"
 	rbacv1 "k8s.io/api/rbac/v1"
 	storagev1 "k8s.io/api/storage/v1"
-	storagebeta "k8s.io/api/storage/v1beta1"
+	storagev1beta1 "k8s.io/api/storage/v1beta1"
 	apierrs "k8s.io/apimachinery/pkg/api/errors"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -1116,7 +1116,7 @@ func getStorageClass(
 }
 
 // TODO: remove when storage.k8s.io/v1beta1 is removed.
-func newBetaStorageClass(t testsuites.StorageClassTest, suffix string) *storagebeta.StorageClass {
+func newBetaStorageClass(t testsuites.StorageClassTest, suffix string) *storagev1beta1.StorageClass {
 	pluginName := t.Provisioner
 
 	if pluginName == "" {
@@ -1126,7 +1126,7 @@ func newBetaStorageClass(t testsuites.StorageClassTest, suffix string) *storageb
 		suffix = "default"
 	}
 
-	return &storagebeta.StorageClass{
+	return &storagev1beta1.StorageClass{
 		TypeMeta: metav1.TypeMeta{
 			Kind: "StorageClass",
 		},