diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index ae2899e..1eaa98f 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -1,8 +1,8 @@
-add_library(neural_amp_modeler MODULE
-	nam_lv2.cpp
+set(SOURCES nam_lv2.cpp
 	nam_plugin.cpp
-	nam_plugin.h
-	../deps/NeuralAmpModelerCore/NAM/activations.h
+	nam_plugin.h)
+
+set(NAM_SOURCES	../deps/NeuralAmpModelerCore/NAM/activations.h
 	../deps/NeuralAmpModelerCore/NAM/version.h
 	../deps/NeuralAmpModelerCore/NAM/lstm.h
 	../deps/NeuralAmpModelerCore/NAM/lstm.cpp
@@ -12,8 +12,12 @@ add_library(neural_amp_modeler MODULE
 	../deps/NeuralAmpModelerCore/NAM/util.cpp
 	../deps/NeuralAmpModelerCore/NAM/util.h
 	../deps/NeuralAmpModelerCore/NAM/wavenet.cpp
-	../deps/NeuralAmpModelerCore/NAM/wavenet.h
-)
+	../deps/NeuralAmpModelerCore/NAM/wavenet.h)
+
+add_library(neural_amp_modeler MODULE ${SOURCES} ${NAM_SOURCES})
+
+source_group(TREE ${CMAKE_CURRENT_SOURCE_DIR} FILES ${SOURCES})
+source_group(NAM ${CMAKE_CURRENT_SOURCE_DIR} FILES ${NAM_SOURCES})
 
 target_compile_features(neural_amp_modeler PUBLIC cxx_std_17)