diff --git a/query_test.go b/query_test.go
index 135805a7..77449f4f 100644
--- a/query_test.go
+++ b/query_test.go
@@ -2,7 +2,6 @@ package gorm_test
 
 import (
 	"fmt"
-	"os"
 	"reflect"
 
 	"github.com/jinzhu/gorm"
@@ -683,7 +682,7 @@ func TestPluckWithSelect(t *testing.T) {
 		combineUserAgeSQL = fmt.Sprintf("concat(%v, %v)", DB.Dialect().Quote("name"), DB.Dialect().Quote("age"))
 	)
 
-	if dialect := os.Getenv("GORM_DIALECT"); dialect == "sqlite" {
+	if dialect := DB.Dialect().GetName(); dialect == "sqlite3" {
 		combineUserAgeSQL = fmt.Sprintf("(%v || %v)", DB.Dialect().Quote("name"), DB.Dialect().Quote("age"))
 	}