diff --git a/pydra/fileformats-medimage-mrtrix3/fileformats/medimage_mrtrix3/__init__.py b/pydra/fileformats-medimage-mrtrix3/fileformats/medimage_mrtrix3/__init__.py
index 9eaff7e53e..1d38dcc3b5 100644
--- a/pydra/fileformats-medimage-mrtrix3/fileformats/medimage_mrtrix3/__init__.py
+++ b/pydra/fileformats-medimage-mrtrix3/fileformats/medimage_mrtrix3/__init__.py
@@ -1,6 +1,6 @@
 from ._version import __version__
 
-from .image import ImageFormat, ImageHeader
+from .image import ImageFormat, ImageFormatGz, ImageHeader, ImageDataFile
 from .in_out import ImageIn, ImageOut
 from .dwi import (
     BFile, NiftiB, NiftiGzB, NiftiGzXB, NiftiXB, ImageFormatB, ImageFormatGzB, ImageHeaderB