diff --git a/js/brightbeam.js b/js/brightbeam.js
index 296e2f9..5afb513 100644
--- a/js/brightbeam.js
+++ b/js/brightbeam.js
@@ -385,6 +385,7 @@ const brightbeam = {
document.getElementById('view').innerText = 'Graph';
document.getElementById('vis-graph').classList.remove('hidden');
document.getElementById('vis-list').classList.add('hidden');
+ document.getElementById('graph-legend').classList.remove('hidden');
this.stopGraphView();
let network = await this.getNetwork();
console.log(network);
@@ -435,6 +436,7 @@ const brightbeam = {
document.getElementById('vis-graph').innerHTML = '';
document.getElementById('vis-graph').classList.add('hidden');
document.getElementById('vis-list').classList.remove('hidden');
+ document.getElementById('graph-legend').classList.add('hidden');
let container = document.getElementById('site-list');
const data = await storeChild.getAll();
diff --git a/manifest.json b/manifest.json
index 1e848a5..e0e5c2b 100644
--- a/manifest.json
+++ b/manifest.json
@@ -1,7 +1,7 @@
{
"manifest_version": 2,
"name": "Brightbeam",
- "version": "4.0.0",
+ "version": "4.0.1",
"description": "Brightbeam is a browser extension that uses interactive visualizations to show you the relationships between third parties and the sites you visit. It is a fork of the late Mozilla Lightbeam.",
"homepage_url": "https://github.com/digitalmethodsinitiative/brightbeam",