diff --git a/README.rst b/README.rst
index 74c729083..9746f12df 100644
--- a/README.rst
+++ b/README.rst
@@ -40,6 +40,7 @@ Documentation_ / Paper_ / Installation_ / Usage_ / Citation_ / Contact_
 
 
 .. _Installation:
+
 Installation
 ====================================================================
 
@@ -71,6 +72,7 @@ directory because otherwise not the in site-packages installed version will be u
 
 
 .. _Usage:
+
 Usage
 ==================================
 
@@ -109,6 +111,7 @@ A representative run of NSGA2 looks as follows:
 
 
 .. _Citation:
+
 Citation
 ====================================================================
 
@@ -129,6 +132,7 @@ Meanwhile, if you have used our framework for research purposes, please cite us
 
 
 .. _Contact:
+
 Contact
 ====================================================================
 
diff --git a/setup.py b/setup.py
index 007373a1f..9d21b44aa 100644
--- a/setup.py
+++ b/setup.py
@@ -63,6 +63,7 @@ def packages():
 
 
 data['long_description'] = readme()
+data['long_description_content_type'] = 'text/x-rst'
 data['packages'] = packages()
 
 
diff --git a/tests/visualization/test_visualization.py b/tests/visualization/test_visualization.py
index 5c36ac0f3..02f81e98d 100644
--- a/tests/visualization/test_visualization.py
+++ b/tests/visualization/test_visualization.py
@@ -9,7 +9,7 @@ class VisualizationTest(unittest.TestCase):
 
     def test_usages(self):
         folder = os.path.join(get_pymoo(), "pymoo", "usage", "visualization")
-        test_usage([os.path.join(folder, fname) for fname in os.listdir(folder)])
+        test_usage([os.path.join(folder, fname) for fname in os.listdir(folder) if fname.endswith("py")])
 
 if __name__ == '__main__':
     unittest.main()