diff --git a/src/deep_neurographs/swc_utils.py b/src/deep_neurographs/swc_utils.py
index 2260e26..d9c2600 100644
--- a/src/deep_neurographs/swc_utils.py
+++ b/src/deep_neurographs/swc_utils.py
@@ -359,7 +359,7 @@ def make_entry(graph, i, parent, node_to_idx):
     ...
 
     """
-    r = set_radius(graph, i)
+    r = graph[i]["radius"]
     x, y, z = tuple(graph.nodes[i]["xyz"])
     node_to_idx[i] = len(node_to_idx)
     entry = f"{node_to_idx[i]} 2 {x} {y} {z} {r} {node_to_idx[parent]}"
diff --git a/src/deep_neurographs/visualization.py b/src/deep_neurographs/visualization.py
index 1442a0a..c956d25 100644
--- a/src/deep_neurographs/visualization.py
+++ b/src/deep_neurographs/visualization.py
@@ -142,7 +142,6 @@ def plot(data, title):
     fig.update_layout(
         title=title,
         template="plotly_white",
-        # plot_bgcolor="rgba(0, 0, 0, 0)",
         scene=dict(aspectmode="manual", aspectratio=dict(x=1, y=1, z=1)),
         width=1200,
         height=700,