Fix tests

Author: munterfi

tests/visualization/test_plot.py

@@ -6,4 +6,4 @@
 def test_generate_plots():
     response = import_response(PkgDataAccess.locate_response())
     figs = generate_plots(response, "test_instance")
-    assert len(figs) == 4
+    assert len(figs) == 6