diff --git a/src/core/partition.h b/src/core/partition.h
index fb54517..576e6ec 100644
--- a/src/core/partition.h
+++ b/src/core/partition.h
@@ -34,6 +34,7 @@ class OperationStack;
 class PartResizerWidget;
 class InsertDialog;
 class NewDialog;
+class EditMountPointDialog;
 
 class CreateFileSystemOperation;
 class RestoreOperation;
@@ -73,6 +74,7 @@ class Partition : public PartitionNode
 	friend class PartResizerWidget;
 	friend class InsertDialog;
 	friend class NewDialog;
+	friend class EditMountPointDialog;
 
 	friend class CreateFileSystemOperation;
 	friend class RestoreOperation;