diff --git a/graph_dit/naswot/naswot/datasets/data.py b/graph_dit/naswot/naswot/datasets/data.py
index e9d8959..aef3651 100644
--- a/graph_dit/naswot/naswot/datasets/data.py
+++ b/graph_dit/naswot/naswot/datasets/data.py
@@ -1,5 +1,5 @@
-from datasets import get_datasets
-from config_utils import load_config
+from naswot.datasets import get_datasets
+from naswot.config_utils import load_config
 import torch
 import torchvision
 
diff --git a/graph_dit/naswot/naswot/utils.py b/graph_dit/naswot/naswot/utils.py
index 800f42f..ed8bed5 100644
--- a/graph_dit/naswot/naswot/utils.py
+++ b/graph_dit/naswot/naswot/utils.py
@@ -1,5 +1,5 @@
 import torch
-from pycls.models.nas.nas import Cell
+from naswot.pycls.models.nas.nas import Cell
 
 class DropChannel(torch.nn.Module):
     def __init__(self, p, mod):