diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index a5cab2a..19e3ac5 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -166,7 +166,7 @@ jobs:
 
       - name: test gpu
         if: always() && !cancelled()
-        run: python tests/gpu.py
+        run: python tests/xpu.py
 
   m4:
     runs-on:  [self-hosted, m4]
diff --git a/tests/gpu.py b/tests/gpu.py
index cf7884b..8a72e6c 100644
--- a/tests/gpu.py
+++ b/tests/gpu.py
@@ -1,4 +1,5 @@
 from device_smi import Device
+import torch
 
 dev = Device("gpu")
 print(dev)