diff --git a/mastodon_reader/utils.py b/mastodon_reader/utils.py
index 44a9df4..2edf001 100644
--- a/mastodon_reader/utils.py
+++ b/mastodon_reader/utils.py
@@ -1,6 +1,6 @@
 import numpy as np
 import networkx as nx
-from matplotlib import pyplot as plt
+
 
 __all__ = ["ismember", "RGBint2RGB"]
 
@@ -229,10 +229,7 @@ def node_positions(self):
 
         return pos
 
-    def draw(self, ax=None):
-        if ax is None:
-            f, ax = plt.subplots()
-
+    def draw(self, ax):
         nx.draw_networkx(
             self.nx_tree,
             pos=self.node_positions(),
diff --git a/setup.py b/setup.py
index b2d3d4c..f34c9aa 100644
--- a/setup.py
+++ b/setup.py
@@ -28,6 +28,5 @@
     #    entry_points = {'console_scripts': []},
     author="Christoph Sommer",
     author_email="christoph.sommer23@gmail.com",
-    install_requires=["numpy", "pandas", "networkx", "matplotlib"],
+    install_requires=["numpy", "pandas", "networkx"],
 )
-