diff --git a/gorm_test.go b/gorm_test.go
index ddc1b81a..503f08e0 100644
--- a/gorm_test.go
+++ b/gorm_test.go
@@ -1,6 +1,8 @@
 package gorm
 
 import (
+	_ "github.com/lib/pq"
+
 	"errors"
 	"reflect"
 	"strconv"
diff --git a/main.go b/main.go
index 0f8d658b..6d4f1870 100644
--- a/main.go
+++ b/main.go
@@ -1,9 +1,6 @@
 package gorm
 
-import (
-	"database/sql"
-	_ "github.com/lib/pq"
-)
+import "database/sql"
 
 type DB struct {
 	Db     *sql.DB
@@ -13,7 +10,6 @@ type DB struct {
 func Open(driver, source string) (db DB, err error) {
 	db.Db, err = sql.Open(driver, source)
 	db.Driver = driver
-	// SetMaxIdleConns pools
 	return
 }