diff --git a/db/test_config.go b/db/test_config.go
index 9adbca6db..ad294609a 100644
--- a/db/test_config.go
+++ b/db/test_config.go
@@ -61,6 +61,8 @@ func InitTestDB() {
 	db.AutoMigrate(&WfRequest{})
 	db.AutoMigrate(&WfProcessingMap{})
 	db.AutoMigrate(&Tickets{})
+	db.AutoMigrate(&ChatMessage{})
+	db.AutoMigrate(&Chat{})
 
 	people := TestDB.GetAllPeople()
 	for _, p := range people {