diff --git a/internal/cmd/db_invalidatetokens.go b/internal/cmd/db_invalidatetokens.go
index 8f162e48..702d7e51 100644
--- a/internal/cmd/db_invalidatetokens.go
+++ b/internal/cmd/db_invalidatetokens.go
@@ -34,8 +34,8 @@ var dbInvalidateTokensCmd = &cobra.Command{
 			return err
 		}
 
-		if database.Group != "" && database.Version != "tech-preview" {
-			return fmt.Errorf("database %s is part of group %s, use %s instead", internal.Emph(name), internal.Emph(database.Group), internal.Emph("turso group tokens invalidate <group-name>"))
+		if database.Group != "" && database.ServerType != "turso-server" {
+			return fmt.Errorf("database %s on sqld server is part of group %s, use %s instead", internal.Emph(name), internal.Emph(database.Group), internal.Emph("turso group tokens invalidate <group-name>"))
 		}
 
 		if yesFlag {
@@ -76,7 +76,7 @@ func rotateAndNotify(turso *turso.Client, database turso.Database) error {
 func rotate(turso *turso.Client, database turso.Database) error {
 	invalidateDbTokenCache()
 	settings.PersistChanges()
-	if database.Group != "" && database.Version != "tech-preview" {
+	if database.Group != "" && database.ServerType != "turso-server" {
 		return turso.Groups.Rotate(database.Group)
 	}
 	return turso.Databases.Rotate(database.Name)
diff --git a/internal/turso/databases.go b/internal/turso/databases.go
index 6726a275..c78b536e 100644
--- a/internal/turso/databases.go
+++ b/internal/turso/databases.go
@@ -28,6 +28,7 @@ type Database struct {
 	IsSchema         bool      `json:"is_schema" mapstructure:"is_schema"`
 	Parent           *Database `json:"parent,omitempty"`
 	EncryptionCipher string    `json:"encryption_cipher,omitempty"`
+	ServerType       string    `json:"server_type"`
 }
 
 type DatabasesClient client