diff --git a/test/torchscript_consistency_cuda_test.py b/test/torchscript_consistency_cuda_test.py
index 49ac5a64ddf..afc09010ee0 100644
--- a/test/torchscript_consistency_cuda_test.py
+++ b/test/torchscript_consistency_cuda_test.py
@@ -1,7 +1,7 @@
 import torch
 
 from . import common_utils
-from .torchscript_consistency_impl import Functional, Transforms
+from .torchscript_consistency_impl import Functional, Transforms, TransformsWithComplexDtypes
 
 
 @common_utils.skipIfNoCuda