diff --git a/animals/models/model_animals.py b/animals/models/model_animals.py
index 4d872a6..988e2bc 100644
--- a/animals/models/model_animals.py
+++ b/animals/models/model_animals.py
@@ -13,7 +13,7 @@
 import torch
 import torch.nn as nn
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 from fmpose3d.animals.models.graph_frames import Graph
 
diff --git a/fmpose3d/animals/models/model_animal3d.py b/fmpose3d/animals/models/model_animal3d.py
index 7ea8b95..68dcbfc 100644
--- a/fmpose3d/animals/models/model_animal3d.py
+++ b/fmpose3d/animals/models/model_animal3d.py
@@ -13,7 +13,7 @@
 import torch
 import torch.nn as nn
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 from fmpose3d.animals.models.graph_frames import Graph
 from fmpose3d.models.base_model import BaseModel, register_model
diff --git a/fmpose3d/lib/hrnet/lib/models/pose_hrnet.py b/fmpose3d/lib/hrnet/lib/models/pose_hrnet.py
index d62d787..a0d0b6a 100755
--- a/fmpose3d/lib/hrnet/lib/models/pose_hrnet.py
+++ b/fmpose3d/lib/hrnet/lib/models/pose_hrnet.py
@@ -493,7 +493,7 @@ def init_weights(self, pretrained=''):
             need_init_state_dict = {}
             for name, m in pretrained_state_dict.items():
                 if name.split('.')[0] in self.pretrained_layers \
-                   or self.pretrained_layers[0] is '*':
+                   or self.pretrained_layers[0] == '*':
                     need_init_state_dict[name] = m
             self.load_state_dict(need_init_state_dict, strict=False)
         elif pretrained:
diff --git a/fmpose3d/models/fmpose3d/model_GAMLP.py b/fmpose3d/models/fmpose3d/model_GAMLP.py
index 7176e26..e4058c0 100644
--- a/fmpose3d/models/fmpose3d/model_GAMLP.py
+++ b/fmpose3d/models/fmpose3d/model_GAMLP.py
@@ -15,7 +15,7 @@
 from fmpose3d.models.base_model import BaseModel, register_model
 from functools import partial
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 class TimeEmbedding(nn.Module):
     def __init__(self, dim: int, hidden_dim: int = 64):