1 files changed, 4 insertions, 7 deletions
diff --git a/golang/pkg/database/tokens.go b/golang/pkg/database/tokens.go
index 16bda2b..0787b5c 100644
--- a/golang/pkg/database/tokens.go
+++ b/golang/pkg/database/tokens.go
@@ -1,14 +1,11 @@
 package database
 
-func (db DB) AddToken(token string) error {
+func (db *DB) AddToken(token string) error {
 	_, err := db.db.ExecContext(db.ctx, `INSERT INTO tokens(data) VALUES (?);`, token)
 	return err
 }
 
-func (db DB) GetToken() (string, error) {
-	var token string
-	if err := db.db.QueryRowContext(db.ctx, `SELECT data FROM tokens;`).Scan(&token); err != nil {
-		return "", err
-	}
-	return token, nil
+func (db *DB) GetToken() (token string, err error) {
+	err = db.db.QueryRowContext(db.ctx, `SELECT data FROM tokens;`).Scan(&token)
+	return token, err
 }