diff --git a/deepmd/infer/data_modifier.py b/deepmd/infer/data_modifier.py
index 2da80c572b..51b5a1f438 100644
--- a/deepmd/infer/data_modifier.py
+++ b/deepmd/infer/data_modifier.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 
+import deepmd.op  # noqa: F401
 from deepmd.common import (
     make_default_mesh,
     select_idx_map,