diff --git a/tests/test_grad.py b/tests/test_grad.py
index 9913f4b..c77b0b9 100644
--- a/tests/test_grad.py
+++ b/tests/test_grad.py
@@ -1,6 +1,6 @@
 import pytest
 import torch
-from torch.autograd import gradcheck, gradgradcheck
+from torch.autograd.gradcheck import gradcheck, gradgradcheck
 from torchlpc.core import LPC