diff --git a/Detail_Model_Patch_Custom.py b/Detail_Model_Patch_Custom.py
index 434ff8b..c58dfe8 100755
--- a/Detail_Model_Patch_Custom.py
+++ b/Detail_Model_Patch_Custom.py
@@ -1,4 +1,4 @@
-zimport torch
+import torch
 import torch.nn as nn
 
 class ModelWrapper(nn.Module):