diff --git a/custom_net/model.py b/custom_net/model.py
index be93a7f..7ad0482 100644
--- a/custom_net/model.py
+++ b/custom_net/model.py
@@ -5,7 +5,7 @@ import torch.nn as nn
 import numpy as np
 from einops import rearrange
 
-from customnet_util import instantiate_from_config
+from .customnet_util import instantiate_from_config
 from .attention import LinearAttention