diff --git a/tests/test_dicom.py b/tests/test_dicom.py
index 9464978..77c2e63 100644
--- a/tests/test_dicom.py
+++ b/tests/test_dicom.py
@@ -8,9 +8,9 @@ def test_dicom_creation(dicom_module):
     out_dir = Path(tempfile.mkdtemp())
 
     # Generate DICOM files
-    dicom_module.get_image(out_dir)
+    dicom_dir = dicom_module.get_image(out_dir)
 
-    assert len(list(out_dir.glob('*.dcm'))) == dicom_module.num_vols
+    assert len(list(dicom_dir.glob("*.dcm"))) == dicom_module.num_vols
 
-    for p in out_dir.glob('*.dcm'):
+    for p in dicom_dir.glob("*.dcm"):
         pydicom.dcmread(p)  # Check dicom file can be read