diff --git a/dataset/craft_datasets.py b/dataset/craft_datasets.py
index b0ebb36..0512926 100644
--- a/dataset/craft_datasets.py
+++ b/dataset/craft_datasets.py
@@ -2,7 +2,6 @@
 import time
 import tensorflow as tf
 import numpy as np
-import borders
 import os
 import sys
 import inspect
@@ -13,6 +12,7 @@
 
 from tools import resize_3d
 import config as config
+from dataset import borders
 
 
 DEBUG_DATALOADER = False