diff --git a/modelci/types/type_conversion.py b/modelci/types/type_conversion.py
index b8471ec2..d7e4bbad 100644
--- a/modelci/types/type_conversion.py
+++ b/modelci/types/type_conversion.py
@@ -102,7 +102,7 @@ def model_data_type_to_np(model_dtype):
 
 
 def model_data_type_to_torch(model_dtype):
-    from modelci.types.bo import DataType
+    from modelci.types.models.common import DataType
     import torch
 
     mapper = {