diff --git a/vista3d/scripts/train.py b/vista3d/scripts/train.py
index e3a1332..0c89d16 100644
--- a/vista3d/scripts/train.py
+++ b/vista3d/scripts/train.py
@@ -40,7 +40,6 @@
 from monai.bundle.scripts import _pop_args, _update_args
 from monai.data import DataLoader, DistributedSampler, DistributedWeightedRandomSampler
 from monai.metrics import compute_dice
-from monai.networks.utils import copy_model_state
 from monai.utils import optional_import, set_determinism
 from torch.nn.parallel import DistributedDataParallel
 from torch.utils.data.sampler import RandomSampler, WeightedRandomSampler
diff --git a/vista3d/scripts/train_finetune.py b/vista3d/scripts/train_finetune.py
index 9a59735..02a5c15 100644
--- a/vista3d/scripts/train_finetune.py
+++ b/vista3d/scripts/train_finetune.py
@@ -35,7 +35,6 @@
 from monai.bundle.scripts import _pop_args, _update_args
 from monai.data import DataLoader, DistributedSampler
 from monai.metrics import compute_dice
-from monai.networks.utils import copy_model_state
 from monai.utils import set_determinism
 from torch.nn.parallel import DistributedDataParallel
 from torch.utils.tensorboard import SummaryWriter