diff --git a/src/leibnetz/__init__.py b/leibnetz/__init__.py
similarity index 100%
rename from src/leibnetz/__init__.py
rename to leibnetz/__init__.py
diff --git a/src/leibnetz/leibnet.py b/leibnetz/leibnet.py
similarity index 100%
rename from src/leibnetz/leibnet.py
rename to leibnetz/leibnet.py
diff --git a/src/leibnetz/nets/__init__.py b/leibnetz/nets/__init__.py
similarity index 100%
rename from src/leibnetz/nets/__init__.py
rename to leibnetz/nets/__init__.py
diff --git a/src/leibnetz/nets/attentive_scalenet.py b/leibnetz/nets/attentive_scalenet.py
similarity index 100%
rename from src/leibnetz/nets/attentive_scalenet.py
rename to leibnetz/nets/attentive_scalenet.py
diff --git a/src/leibnetz/nets/scalenet.py b/leibnetz/nets/scalenet.py
similarity index 100%
rename from src/leibnetz/nets/scalenet.py
rename to leibnetz/nets/scalenet.py
diff --git a/src/leibnetz/nets/unet.py b/leibnetz/nets/unet.py
similarity index 100%
rename from src/leibnetz/nets/unet.py
rename to leibnetz/nets/unet.py
diff --git a/src/leibnetz/nodes/__init__.py b/leibnetz/nodes/__init__.py
similarity index 100%
rename from src/leibnetz/nodes/__init__.py
rename to leibnetz/nodes/__init__.py
diff --git a/src/leibnetz/nodes/additive_attention_gate_node.py b/leibnetz/nodes/additive_attention_gate_node.py
similarity index 100%
rename from src/leibnetz/nodes/additive_attention_gate_node.py
rename to leibnetz/nodes/additive_attention_gate_node.py
diff --git a/src/leibnetz/nodes/conv_pass_node.py b/leibnetz/nodes/conv_pass_node.py
similarity index 100%
rename from src/leibnetz/nodes/conv_pass_node.py
rename to leibnetz/nodes/conv_pass_node.py
diff --git a/src/leibnetz/nodes/conv_resample_node.py b/leibnetz/nodes/conv_resample_node.py
similarity index 100%
rename from src/leibnetz/nodes/conv_resample_node.py
rename to leibnetz/nodes/conv_resample_node.py
diff --git a/src/leibnetz/nodes/node.py b/leibnetz/nodes/node.py
similarity index 100%
rename from src/leibnetz/nodes/node.py
rename to leibnetz/nodes/node.py
diff --git a/src/leibnetz/nodes/node_ops.py b/leibnetz/nodes/node_ops.py
similarity index 100%
rename from src/leibnetz/nodes/node_ops.py
rename to leibnetz/nodes/node_ops.py
diff --git a/src/leibnetz/nodes/resample_node.py b/leibnetz/nodes/resample_node.py
similarity index 100%
rename from src/leibnetz/nodes/resample_node.py
rename to leibnetz/nodes/resample_node.py
diff --git a/src/leibnetz/nodes/resample_ops.py b/leibnetz/nodes/resample_ops.py
similarity index 100%
rename from src/leibnetz/nodes/resample_ops.py
rename to leibnetz/nodes/resample_ops.py
diff --git a/setup.cfg b/setup.cfg
index e7ea71f..5cde93d 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -15,7 +15,7 @@ classifiers =
 
 [options]
 package_dir=
-    =src
+    =leibnetz
 packages=find:
 zip_safe = True
 include_package_data = True
@@ -25,5 +25,5 @@ install_requires =
     funlib.learn.torch @ git+https://github.com/funkelab/funlib.learn.torch.git
 
 [options.packages.find]
-where=src
+where=leibnetz
 
diff --git a/src/leibnetz/tests/__init__.py b/tests/__init__.py
similarity index 100%
rename from src/leibnetz/tests/__init__.py
rename to tests/__init__.py
diff --git a/src/leibnetz/tests/additive_attention_gate_node_test.py b/tests/additive_attention_gate_node_test.py
similarity index 100%
rename from src/leibnetz/tests/additive_attention_gate_node_test.py
rename to tests/additive_attention_gate_node_test.py
diff --git a/src/leibnetz/tests/benchmark.py b/tests/benchmark.py
similarity index 98%
rename from src/leibnetz/tests/benchmark.py
rename to tests/benchmark.py
index 2990969..59b028e 100644
--- a/src/leibnetz/tests/benchmark.py
+++ b/tests/benchmark.py
@@ -1,6 +1,6 @@
 # %%
 import torch
-from leibnetz.unet import build_unet as leibnetz_unet
+from leibnetz.nets import build_unet as leibnetz_unet
 from funlib.learn.torch.models import UNet as funlib_unet
 # set flag to improve training speeds
 torch.backends.cudnn.benchmark = True
diff --git a/src/leibnetz/tests/conv_pass_node_test.py b/tests/conv_pass_node_test.py
similarity index 100%
rename from src/leibnetz/tests/conv_pass_node_test.py
rename to tests/conv_pass_node_test.py
diff --git a/src/leibnetz/tests/leibnet_test.py b/tests/leibnet_test.py
similarity index 100%
rename from src/leibnetz/tests/leibnet_test.py
rename to tests/leibnet_test.py