From 397671a9e45048fca627ced6139020d752ac4d91 Mon Sep 17 00:00:00 2001
From: oyelowo <oyelowooyedayo@gmail.com>
Date: Thu, 18 Jan 2024 06:13:22 -0800
Subject: [PATCH] Fix MlModel statement usage

---
 migrator/src/database/queries.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/migrator/src/database/queries.rs b/migrator/src/database/queries.rs
index 5cdcd4cd8b..beef72b47b 100644
--- a/migrator/src/database/queries.rs
+++ b/migrator/src/database/queries.rs
@@ -274,7 +274,7 @@ impl DefineStatementRaw {
                     };
                     remove_stmt.to_raw().build()
                 }
-                DefineStatement::MlModel(ml) => remove_model(get_resource_name(ml.name.to_raw()))
+                DefineStatement::Model(ml) => remove_model(get_resource_name(ml.name.to_raw()))
                     .version(ml.version)
                     .to_raw()
                     .build(),