diff --git a/afqinsight/nn/utils.py b/afqinsight/nn/utils.py
index 1692f34..34ffb6f 100644
--- a/afqinsight/nn/utils.py
+++ b/afqinsight/nn/utils.py
@@ -2,11 +2,12 @@
 import tensorflow as tf
 import torch
 import torch.nn as nn
-from neurocombat_sklearn import CombatModel
 from sklearn.impute import SimpleImputer
 from sklearn.model_selection import train_test_split
 from tensorflow.keras import layers
 
+from afqinsight.neurocombat_sklearn import CombatModel
+
 
 def extract_layer_info_pytorch(layer):
     info = {}