diff --git a/pipelines/floret_ko_ud_demo/scripts/train_floret.py b/pipelines/floret_ko_ud_demo/scripts/train_floret.py
index ee9ac6c30..b29136e68 100644
--- a/pipelines/floret_ko_ud_demo/scripts/train_floret.py
+++ b/pipelines/floret_ko_ud_demo/scripts/train_floret.py
@@ -1,6 +1,7 @@
 import typer
+import numpy
 from pathlib import Path
-import floret
+
 
 
 def main(