diff --git a/examples/tree1024.py b/examples/tree1024.py
index 2fb001d..4684d2d 100755
--- a/examples/tree1024.py
+++ b/examples/tree1024.py
@@ -9,18 +9,11 @@ and running sysctl -p. Check util/sysctl_addon.
 
 from mininet.cli import CLI
 from mininet.log import setLogLevel
-from mininet.net import init, Mininet
 from mininet.node import KernelSwitch
-from mininet.topolib import TreeTopo
-
-def TreeNet( depth=1, fanout=2, **kwargs ):
-    "Convenience function for creating tree networks."
-    topo = TreeTopo( depth, fanout )
-    return Mininet( topo, **kwargs )
+from mininet.topolib import TreeNet
 
 if __name__ == '__main__':
     setLogLevel( 'info' )
-    init()
-    KernelSwitch.setup()
     network = TreeNet( depth=2, fanout=32, switch=KernelSwitch )
     network.run( CLI, network )
+