diff --git a/generation/3d_ldm/inference.py b/generation/3d_ldm/inference.py
index 6786dd2ba..19c638a67 100644
--- a/generation/3d_ldm/inference.py
+++ b/generation/3d_ldm/inference.py
@@ -20,8 +20,8 @@
 import nibabel as nib
 import numpy as np
 import torch
-from generative.inferers import LatentDiffusionInferer
-from generative.networks.schedulers import DDPMScheduler
+from monai.inferers import LatentDiffusionInferer
+from monai.networks.schedulers import DDPMScheduler
 from monai.config import print_config
 from monai.utils import set_determinism