diff --git a/opacus/grad_sample/linear.py b/opacus/grad_sample/linear.py
index 2cdb84cd..cceb8cac 100644
--- a/opacus/grad_sample/linear.py
+++ b/opacus/grad_sample/linear.py
@@ -18,7 +18,7 @@
 
 import torch
 import torch.nn as nn
-from opt_einsum import contract
+from opt_einsum.contract import contract
 
 from .utils import register_grad_sampler, register_norm_sampler