(feat) fix test db query