diff --git a/src/brevitas/proxy/parameter_quant.py b/src/brevitas/proxy/parameter_quant.py
index c701efac5..6a3e024e2 100644
--- a/src/brevitas/proxy/parameter_quant.py
+++ b/src/brevitas/proxy/parameter_quant.py
@@ -8,6 +8,7 @@
 from warnings import warn
 
 import packaging.version
+import torch
 
 from brevitas import torch_version
 
@@ -16,7 +17,6 @@
 else:
     is_dynamo_compiling = torch._dynamo.is_compiling
 
-import torch
 from torch import Tensor
 import torch.nn as nn
 from typing_extensions import Protocol