From abef79ada3482bcb0159f920ea911d6e967a82be Mon Sep 17 00:00:00 2001
From: Jakub Novak <jakub@e2b.dev>
Date: Thu, 16 Jan 2025 09:53:14 -0800
Subject: [PATCH] Fix seed script

---
 packages/shared/scripts/seed-db.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/shared/scripts/seed-db.go b/packages/shared/scripts/seed-db.go
index a55ae2ed9..5242f5132 100644
--- a/packages/shared/scripts/seed-db.go
+++ b/packages/shared/scripts/seed-db.go
@@ -75,11 +75,16 @@ func main() {
 	}
 
 	// Create team
-	t, err := database.Client.Team.Create().SetEmail(email).SetName("E2B").SetID(teamUUID).SetTier("base_v1").AddUsers(user).Save(ctx)
+	t, err := database.Client.Team.Create().SetEmail(email).SetName("E2B").SetID(teamUUID).SetTier("base_v1").Save(ctx)
 	if err != nil {
 		panic(err)
 	}
 
+	// Create user team
+	_, err = database.Client.UsersTeams.Create().SetUserID(user.ID).SetTeamID(t.ID).SetIsDefault(true).Save(ctx)
+	if err != nil {
+		panic(err)
+	}
 	// Create access token
 	_, err = database.Client.AccessToken.Create().SetUser(user).SetID(accessToken).Save(ctx)
 	if err != nil {