diff --git a/main/SMPLer_X.py b/main/SMPLer_X.py
index 86fad6a..57e7e03 100644
--- a/main/SMPLer_X.py
+++ b/main/SMPLer_X.py
@@ -8,7 +8,7 @@ from utils.transforms import rot6d_to_axis_angle, restore_bbox
 from config import cfg
 import math
 import copy
-from mmposesmplerx.models import build_posenet
+from mmpose_smplerx.models import build_posenet
 from mmengine.config import Config
 
 class Model(nn.Module):
diff --git a/main/transformer_utils/mmpose_smplerx/models/builder.py b/main/transformer_utils/mmpose_smplerx/models/builder.py
index 5fa61ca..a1f909f 100644
--- a/main/transformer_utils/mmpose_smplerx/models/builder.py
+++ b/main/transformer_utils/mmpose_smplerx/models/builder.py
@@ -3,7 +3,7 @@ from mmengine.registry import MODELS as MMCV_MODELS
 from mmengine import Registry
 from mmengine.registry import build_from_cfg, build_model_from_cfg
 
-MODELS = Registry('models', parent=MMCV_MODELS, locations=['mmpose.models'])
+MODELS = Registry('models', parent=MMCV_MODELS, locations=['mmpose_smplerx.models'])
 
 BACKBONES = MODELS
 NECKS = MODELS