diff --git a/setup.py b/setup.py
index db6741fd0f8d6f8a90a6d67796a1d5c985a8a54c..f9bbb936cb62b2a8ede3813c096137bae54a016d 100644
--- a/setup.py
+++ b/setup.py
@@ -32,5 +32,7 @@ setup(name="nifty",
       package_dir={"nifty": ""},
       package_data={"nifty": ["demos/demo_excaliwir.py",
                               "demos/demo_faraday.py",
-                              "demos/demo_faraday_map.npy"]},
+                              "demos/demo_faraday_map.npy",
+                              "demos/demo_wf1.py",
+                              "demos/demo_wf2.py"]},
       data_files=[(os.path.expanduser('~') + "/.nifty", ["nifty_config"])])
\ No newline at end of file