diff --git a/py/torch_tensorrt/ts/ptq.py b/py/torch_tensorrt/ts/ptq.py
index ec86c620ea..da2fbee47b 100644
--- a/py/torch_tensorrt/ts/ptq.py
+++ b/py/torch_tensorrt/ts/ptq.py
@@ -11,7 +11,7 @@
 
 import torch
 from torch_tensorrt import _C
-from py.torch_tensorrt.ts.logging import Level, log
+from torch_tensorrt.ts.logging import Level, log
 
 
 class CalibrationAlgo(Enum):