diff --git a/setup.py b/setup.py
index 5dfdbbf..df34f81 100644
--- a/setup.py
+++ b/setup.py
@@ -53,9 +53,9 @@
           'Programming Language :: Python :: 2',
           'Programming Language :: Python :: 3',
           ],
-      packages=['xija', 'xija.component', 'xija.tests', 'xija.gui_fit'],
+      packages=['xija', 'xija.component', 'xija.component.heat', 'xija.tests', 'xija.gui_fit'],
       package_data={'xija': ['libcore.so',
-                             'component/earth_vis_grid_nside32.fits.gz'],
+                             'component/heat/earth_vis_grid_nside32.fits.gz'],
                     'xija.tests': ['*.npz', '*.json'],
                     'xija.gui_fit': ['app_icon.png']},
       tests_require=['pytest'],