diff --git a/nbodykit/lab.py b/nbodykit/lab.py
index ee4e3ee8d..2314a7a71 100644
--- a/nbodykit/lab.py
+++ b/nbodykit/lab.py
@@ -1,11 +1,15 @@
 from mpi4py import MPI
-
 import numpy
 
-from nbodykit.batch import TaskManager 
-from nbodykit import source as Source, cosmology
+# sources
+from nbodykit.source.catalog import *
+from nbodykit.source.mesh import *
+
+# algorithms
 from nbodykit.algorithms import *
+
+from nbodykit.batch import TaskManager 
+from nbodykit import cosmology
 from nbodykit import CurrentMPIComm
 from nbodykit import transform
-
 from nbodykit import io as IO
\ No newline at end of file