diff --git a/lcviz/tests/test_parser.py b/lcviz/tests/test_parser.py
index 4bac62dc..f70691ac 100644
--- a/lcviz/tests/test_parser.py
+++ b/lcviz/tests/test_parser.py
@@ -5,6 +5,7 @@
 from astropy.utils.data import download_file
 from lightkurve import LightCurve, search_targetpixelfile
 from lightkurve.io import kepler
+from lightkurve.targetpixelfile import TargetPixelFile
 import astropy.units as u
 
 from lcviz.utils import TimeCoordinates
@@ -58,6 +59,7 @@ def test_kepler_tpf_via_lightkurve(helper):
                                  quarter=10).download()
     helper.load_data(tpf)
     assert helper.get_data().shape == (4447, 4, 6)  # (time, x, y)
+    assert helper.app.data_collection[0].get_object(cls=TargetPixelFile).shape == (4447, 4, 6)
 
 
 def test_synthetic_lc(helper):