diff --git a/main_test.go b/main_test.go
index 666ba564..0dc5e337 100644
--- a/main_test.go
+++ b/main_test.go
@@ -66,6 +66,7 @@ func TestStringPrimaryKey(t *testing.T) {
 		ID   string `gorm:"primary_key"`
 		Name string
 	}
+	DB.AutoMigrate(&UUIDStruct{})
 
 	data := UUIDStruct{ID: "uuid", Name: "hello"}
 	if err := DB.Save(&data).Error; err != nil || data.ID != "uuid" {