diff --git a/depthfm/unet/openaimodel.py b/depthfm/unet/openaimodel.py
index 69eadf1..6f869af 100644
--- a/depthfm/unet/openaimodel.py
+++ b/depthfm/unet/openaimodel.py
@@ -5,7 +5,7 @@ import torch.nn as nn
 from abc import abstractmethod
 import torch.nn.functional as F
 
-from util import (
+from .util import (
     checkpoint,
     conv_nd,
     linear,