diff --git a/database/connect.go b/database/connect.go
index 408f9d1..ca8347d 100644
--- a/database/connect.go
+++ b/database/connect.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 
+	"github.com/jackc/pgx/v5"
 	"github.com/jackc/pgx/v5/pgconn"
 	"github.com/jackc/pgx/v5/pgxpool"
 )
@@ -27,17 +28,10 @@ func IsDuplicatedRow(err error) error {
 	return nil
 }
 
-// FIXME get an idea how this func should work
-// func NoRowsInQuerySet(err error) error {
-// 	var pgErr *pgconn.PgError
+func NoRowsInQuerySet(err error) error {
+	if err == pgx.ErrNoRows {
+		return errors.New("no rows found")
+	}
 
-// 	if errors.As(err, &pgErr) {
-// 		fmt.Println("pgerr:", pgErr.Code)
-// 		return errors.New("no rows selected")
-// 	}
-// 	// if err == pgconn. {
-// 	// 	fmt.Println("No rows were found.")
-// 	// }
-
-// 	return nil
-// }
+	return nil
+}