diff --git a/setup.py b/setup.py
index 3b2083d..06c698e 100644
--- a/setup.py
+++ b/setup.py
@@ -28,5 +28,5 @@
         'scipy'
     ],
     license='BSD (3 Clause)',
-    extras_require=dict(tests=['pytest'], demos=['matplotlib', 'jupyter'], docs=['sphinx']),
+    extras_require=dict(tests=['pytest'], demos=['matplotlib', 'jupyter'], docs=['sphinx', 'nbsphinx', 'sphinx_rtd_theme']),
 )
diff --git a/tests/test_cmv.py b/tests/test_cmv.py
index 2d7f58e..808d551 100644
--- a/tests/test_cmv.py
+++ b/tests/test_cmv.py
@@ -97,7 +97,7 @@ def test_cmv_artificial(theta_deg, velocity, mode):
 
 
 def test_cmv_gagne_data():
-    datafile = "../demos/data/sample_plant_1.h5"
+    datafile = "../docs/sphinx/source/demos/data/sample_plant_1.h5"
     pos_utm = pd.read_hdf(datafile, mode="r", key="utm")
     df = pd.read_hdf(datafile, mode="r", key="data_a")
 
@@ -115,7 +115,7 @@ def test_cmv_gagne_data():
 
 
 def test_cmv_jamaly_data():
-    datafile = "../demos/data/sample_plant_1.h5"
+    datafile = "../docs/sphinx/source/demos/data/sample_plant_1.h5"
     pos_utm = pd.read_hdf(datafile, mode="r", key="utm")
     df = pd.read_hdf(datafile, mode="r", key="data_a")
 
@@ -133,7 +133,7 @@ def test_cmv_jamaly_data():
 def test_cmv_jamaly_data_ref():
     # A test using a reference when we calculate the CMV
 
-    datafile = "../demos/data/sample_plant_1.h5"
+    datafile = "../docs/sphinx/source/demos/data/sample_plant_1.h5"
     pos_utm = pd.read_hdf(datafile, mode="r", key="utm")
     df = pd.read_hdf(datafile, mode="r", key="data_a")
 
diff --git a/tests/test_field.py b/tests/test_field.py
index a531532..d265f2b 100644
--- a/tests/test_field.py
+++ b/tests/test_field.py
@@ -88,7 +88,7 @@ def test_compute_predicted_position_static(refdat):
     ref = refdat[2][0]
     expect = refdat[2][1]
 
-    datafile = "../demos/data/sample_plant_2.h5"
+    datafile = "../docs/sphinx/source/demos/data/sample_plant_2.h5"
     cmv_a = spatial.pol2rect(9.52, 0.62)
     cmv_b = spatial.pol2rect(8.47, 2.17)