diff --git a/README.md b/README.md
index 7fc88ea..d4e31a0 100644
--- a/README.md
+++ b/README.md
@@ -10,10 +10,18 @@
 2. `conda activate uips`
 3. `pip install .`
 
+Test
+
+```
+bash tutorials/run2D.sh
+```
+
 ## Installation for users
 
 `pip install uips`
 
+Test
+
 ```
 import os
 
diff --git a/tests/main_from_input.py b/tests/main_from_input.py
index 4b3036f..7e2a1bc 100644
--- a/tests/main_from_input.py
+++ b/tests/main_from_input.py
@@ -1,6 +1,6 @@
 import argparse
 
-from uips.wrapper import downsample_dataset_from_input
+from uips.wrapper import downsample_dataset_from_input_file
 
 parser = argparse.ArgumentParser(description="Downsampler")
 parser.add_argument(
diff --git a/uips/wrapper.py b/uips/wrapper.py
index b2835bf..9f33df9 100644
--- a/uips/wrapper.py
+++ b/uips/wrapper.py
@@ -220,9 +220,9 @@ def downsample_dataset_from_input(inpt):
             )
             data_for_pdf_est = downSampledData
 
+    best_files = {}
     # Advise for best sampling
     if par.irank == par.iroot:
-        best_files = {}
         if np.amax(meanCriterion) > 0:
             print("\n")
             maxCrit = np.argmax(meanCriterion, axis=0)