diff --git a/custom_net/openaimodel.py b/custom_net/openaimodel.py
index f75a065..2e975dc 100644
--- a/custom_net/openaimodel.py
+++ b/custom_net/openaimodel.py
@@ -18,7 +18,7 @@ from .util import (
     timestep_embedding,
 )
 from .attention import SpatialTransformer
-from customnet_util import exists
+from .customnet_util import exists
 
 
 # dummy replace