diff --git a/test/test_data_pet.py b/test/test_data_pet.py
index 1e40cf8a..6f433bb4 100644
--- a/test/test_data_pet.py
+++ b/test/test_data_pet.py
@@ -29,6 +29,7 @@
 import pytest
 from nitransforms.linear import Affine
 
+from nifreeze.data import load
 from nifreeze.data.pet import PET, _compute_frame_duration, _compute_uptake_statistic, from_nii
 from nifreeze.utils.ndimage import load_api
 
@@ -63,6 +64,12 @@ def random_nifti_file(tmp_path, setup_random_uniform_spatial_data) -> Path:
     return _filename
 
 
+def test_main(datadir):
+    input_file = datadir / "pet.h5"
+
+    assert isinstance(load(input_file), PET)
+
+
 @pytest.mark.parametrize(
     "midframe, expected",
     [