diff --git a/custom_net/openaimodeli.py b/custom_net/openaimodeli.py
index 504ecc2..54ff9a3 100644
--- a/custom_net/openaimodeli.py
+++ b/custom_net/openaimodeli.py
@@ -8,7 +8,7 @@ import torch as th
 import torch.nn as nn
 import torch.nn.functional as F
 
-from customnet_modules.util import (
+from .util import (
     checkpoint,
     conv_nd,
     linear,