diff --git a/scope/nn.py b/scope/nn.py
index 463d81b..849debf 100644
--- a/scope/nn.py
+++ b/scope/nn.py
@@ -16,7 +16,7 @@
 import wandb
 import json
 
-from .models import AbstractClassifier
+from models import AbstractClassifier
 
 
 class DenseBlock(tf.keras.models.Model):
diff --git a/scope/xgb.py b/scope/xgb.py
index 26eb793..f36a273 100644
--- a/scope/xgb.py
+++ b/scope/xgb.py
@@ -1,5 +1,5 @@
 import pathlib
-from .models import AbstractClassifier
+from models import AbstractClassifier
 import xgboost as xgb
 from sklearn.metrics import (
     confusion_matrix,