1 files changed, 4 insertions, 2 deletions
diff --git a/src/drumkit.h b/src/drumkit.h
index 1786584..4cee6be 100644
--- a/src/drumkit.h
+++ b/src/drumkit.h
@@ -35,8 +35,6 @@
 
 class DrumKit
 {
-	friend class DrumKitParser;
-	friend class DrumKitLoader;
 public:
 	DrumKit();
 	~DrumKit();
@@ -60,6 +58,10 @@ public:
 	std::size_t getNumberOfFiles() const;
 
 private:
+	friend class DOMLoader;
+	friend class DOMLoaderTest;
+	friend class DrumKitParser;
+	friend class DrumKitLoader;
 	friend class DrumkitParserTest;
 
 	void* magic{nullptr};