diff --git a/api/graphql.go b/api/graphql.go
index 28cd86e..9754137 100644
--- a/api/graphql.go
+++ b/api/graphql.go
@@ -85,15 +85,14 @@ func generateGraphQLType(model interface{}) (*graphql.Object, error) {
 
 func NewSchema(config Config) (graphql.Schema, error) {
 
-	db, err = gorm.Open(mysql.Open(fmt.Sprintf(
+	if db, err = gorm.Open(mysql.Open(fmt.Sprintf(
 		"%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
 		config.Mysql.UserName,
 		config.Mysql.Password,
 		config.Mysql.Host,
 		config.Mysql.Port,
 		config.Mysql.Database,
-	)), &gorm.Config{})
-	if err != nil {
+	)), &gorm.Config{}); err != nil {
 		log.Fatal("failed to connect to database:", err)
 	}