diff --git a/client.go b/client.go
index 9556751..08ae357 100644
--- a/client.go
+++ b/client.go
@@ -4,7 +4,7 @@ import (
 	"errors"
 	"fmt"
 	"github.com/injoyai/base/maps"
-	"github.com/injoyai/base/maps/wait/v2"
+	"github.com/injoyai/base/maps/wait"
 	"github.com/injoyai/conv"
 	"github.com/injoyai/ios"
 	"github.com/injoyai/ios/client"