diff --git a/custom_net/autoencoder.py b/custom_net/autoencoder.py
index bed2692..8634d6a 100644
--- a/custom_net/autoencoder.py
+++ b/custom_net/autoencoder.py
@@ -8,7 +8,7 @@ from taming.modules.vqvae.quantize import VectorQuantizer
 from .model import Encoder, Decoder
 from .distributions import DiagonalGaussianDistribution
 
-from customnet_util import instantiate_from_config
+from .customnet_util import instantiate_from_config
 
 
 class VQModel(pl.LightningModule):