set Code when inserting

invites/invites.go

@@ -32,7 +32,9 @@ func (i *Invite) Insert(db *sql.DB) error {
 		return err
 	}
 
-	_, err = stmt.Exec(generateCode(i.Email), i.Email)
+	i.Code = generateCode(i.Email)
+
+	_, err = stmt.Exec(i.Code, i.Email)
 	if err != nil {
 		return err
 	}